cuequivariance-torch
von nvidia
Führen Sie äquivariante Tensorprodukte in PyTorch aus, unter Verwendung von SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), High-Level-Operationen…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch