Software Development

cuEquivariance JAX

Use when writing or debugging cuEquivariance JAX polynomials, irrep dictionaries, Flax NNX layers, or CPU/CUDA execution and layouts.

Publisher / source
NVIDIA Skills (Verified)
Upstream record
NVIDIA/skills
Version
Version 1
Published
Publication review
Approved for publication
Content fingerprint
59273471279715c22098fe71801d3abd0ecc8dadc294cc960689d1777596180c