cuequivariance-torch

作成者: nvidia

Execute equivariant tensor products in PyTorch using SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), high-level operations…

npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch