CUDA Equivariant Operations for PyTorch

cuequivariance (version 0.9.1) provides efficient and flexible CUDA kernels for equivariant operations on 3D data, specifically designed for use with PyTorch. It aims to accelerate computations for deep learning models that require rotational, translational, or other forms of equivariance, often found in point cloud processing and molecular modeling. The library is actively developed and part of the vLLM project.