cuequivariance-jax
โดย nvidia
ดำเนินการพหุนามสมมูลใน JAX โดยใช้ segmented_polynomial (naive/uniform_1d), เวิร์กโฟลว์ ir_dict กับ IrDictPolynomial และ dict[Irrep, Array] และ…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax