cuequivariance-torch
от nvidia
Выполнение эквивариантных тензорных произведений в PyTorch с использованием SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), высокоуровневые операции…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch