cuequivariance-torch

作者: nvidia

在PyTorch中使用SegmentedPolynomial(naive/uniform_1d/fused_tp/indexed_linear)執行等變張量積、高階運算…

npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch

來自 nvidia 的更多技能