cuEquivariance
Python
★ 403
updated 9d ago
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.
No plain-English explanation yet — one is being written right now. Check back in a minute.