PASSIONLab/OpenEquivariance
OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.
GitHub repository with 149 stars and 9 forks.
Language: Python
Topics: cuda, geometric-deep-learning, graph-neural-networks, sparse-tensors, equivariance, hip, jax, pytorch