Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
AlphaUncommon

NVIDIA/cuEquivariance

cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffDock, MACE, Allegro and NEQUIP, based on equivariant neural networks. Also includes kernels for accelerated structure prediction.

Owner
NVIDIA (organization)
Stars
Stars: 446
Forks
Forks: 47
Language
Python
License
None
Created
2024
Tier
Alpha (250 to 699 ★) What this tier means
View on GitHub

More repos by NVIDIA

All 269 repos by NVIDIA

Nearby in Python

All Python repos