cuequivariance-torch
bởi nvidia
Thực hiện các tích tensor đẳng biến trong PyTorch bằng SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), các thao tác cấp cao…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch