cuequivariance-jax
por nvidia
Execute polinômios equivariantes em JAX usando segmented_polynomial (naive/uniform_1d), o fluxo de trabalho ir_dict com IrDictPolynomial e dict[Irrep, Array], e…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax