Catalog

Skill by nvidia

cuequivariance-torch

Use when building or debugging cuEquivariance PyTorch tensor products or equivariant GNNs, especially CPU references, layouts, indexing, and gradients.

NVIDIA skillAI EngineerMl EngineerHpc DeveloperApplication DeveloperResearch AcademicAI And Machine Learning