Source

nvidia/cuequivariance

3 skills · 5 combined installs

Skills from this source

#

Skill

Source

8W Activity

Installs

cuequivariance-jax Execute equivariant polynomials in JAX using segmented_polynomial (naive/uniform_1d), the ir_dict workflow with IrDic…

nvidia/cuequivariance

cuequivariance-torch Execute equivariant tensor products in PyTorch using SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), …

nvidia/cuequivariance

cuequivariance Define custom groups (Irrep subclasses), build segmented tensor products with CG coefficients, create equivariant pol…

nvidia/cuequivariance