cuequivariance-jax
oleh nvidia
Jalankan polinomial ekuivarian dalam JAX menggunakan segmented_polynomial (naive/uniform_1d), alur kerja ir_dict dengan IrDictPolynomial dan dict[Irrep, Array], dan…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax