cuequivariance-jax

作者: 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