cuequivariance-torch
द्वारा nvidia
PyTorch में SegmentedPolynomial (naive/uniform_1d/fused_tp/indexed_linear) का उपयोग करके समविभिन्न टेंसर गुणनफल निष्पादित करें, उच्च-स्तरीय संक्रियाएँ…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-torch