cuequivariance-torch
โดย nvidia
ดำเนินการเทนเซอร์โปรดักส์แบบเอกวิเวอร์เรียนต์ใน PyTorch โดยใช้ SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear), การดำเนินการระดับสูง…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch