cuequivariance-jax
von nvidia
Führen Sie äquivariante Polynome in JAX aus, indem Sie segmented_polynomial (naive/uniform_1d), den ir_dict-Workflow mit IrDictPolynomial und dict[Irrep, Array] sowie… verwenden.
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax