cuequivariance-jax
par nvidia
Execute equivariant polynomials in JAX using segmented_polynomial (naive/uniform_1d), the ir_dict workflow with IrDictPolynomial and dict[Irrep, Array], and…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax