cuequivariance-jax
por nvidia
Ejecuta polinomios equivariantes en JAX usando segmented_polynomial (naive/uniform_1d), el flujo de trabajo ir_dict con IrDictPolynomial y dict[Irrep, Array], y…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax