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 stale · no recent hits
total hits 30
actors 8 distinct systems
last hit 17d ago ByteDance
top countries 🇸🇬 Singapore · 🇺🇸 United States · 🇷🇺 Russia · 🇨🇳 China · 🇨🇦 Canada
Resources
API endpoints
full doc /v1/registry/cuequivariance-torch
compatibility /v1/registry/cuequivariance-torch/compatibility