cuequivariance-jax
par nvidia
Exécutez des polynômes équivariants dans JAX en utilisant segmented_polynomial (naive/uniform_1d), le workflow ir_dict avec IrDictPolynomial et dict[Irrep, Array], et…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax