NVIDIA cuEquivariance
NVIDIA cuEquivariance cuEquivariance is a CUDA-X™ library specifically designed to tackle the demanding computational requirements of geometry-aware neural networks, which are essential for tasks involving 3D data. cuEquivariance provides optimized NVIDIA CUDA® kernels and comprehensive APIs, inclu… …