在JAX中使用segmented_polynomial(naive/uniform_1d)、ir_dict工作流搭配IrDictPolynomial與dict[Irrep, Array]來執行等變多項式,以及…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax