JAXで等変多項式を実行する。segmented_polynomial(naive/uniform_1d)、IrDictPolynomialとdict[Irrep, Array]を用いたir_dictワークフロー、および…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax