cuequivariance-torch
tarafından nvidia
PyTorch'da SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear) kullanarak eşdeğişken tensör çarpımları gerçekleştirin, yüksek seviyeli işlemler…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch