cuequivariance-torch
por nvidia
Ejecuta productos tensoriales equivariantes en PyTorch usando SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), operaciones de alto nivel…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch