Catalog
Skill by nvidia
cuequivariance-jax
Use when writing or debugging cuEquivariance JAX polynomials, irrep dictionaries, Flax NNX layers, or CPU/CUDA execution and layouts.
NVIDIA skillAI EngineerMl EngineerHpc DeveloperApplication DeveloperResearch AcademicAI And Machine Learning