CUDA Equivariance for PyTorch

JSON →
library 0.9.1 ·python
verified May 25, 2026

cuequivariance-torch provides CUDA-accelerated implementations of equivariant operations for PyTorch. It aims to efficiently handle geometric symmetries in deep learning models, particularly for 3D data, by offering modules for SO(3) rotations and SE(3) translations. The current version is 0.9.1, and it maintains an active development pace with new features and optimizations.

total hits 38
actors 9 distinct systems
last hit 21h ago human
ByteDance
11
Amazonbot
4
MetaBot
4
GPTBot
2
Script
2
PerplexityBot
1
Search engines
1
Humans
4

top countries 🇸🇬 Singapore · 🇺🇸 United States · 🇩🇪 Germany · 🇨🇦 Canada · VN