cuequivariance-torch
par nvidia
Exécutez des produits tensoriels équivariants dans PyTorch en utilisant SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), des opérations de haut niveau…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch