PyTorch上でSegmentedPolynomial(naive/uniform_1d/fused_tp/indexed_linear)を用いて同変テンソル積を実行し、高レベルな操作を…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch