CUDA Equivariance for PyTorch
JSON →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.
Traffic · last 30 days ↓35% vs prev 7d
total hits 38
actors 9 distinct systems
last hit 21h ago human
top countries 🇸🇬 Singapore · 🇺🇸 United States · 🇩🇪 Germany · 🇨🇦 Canada · VN
Resources
API endpoints
full doc /v1/registry/cuequivariance-torch
compatibility /v1/registry/cuequivariance-torch/compatibility