cuequivariance-jax

作者: nvidia

在JAX中使用segmented_polynomial(naive/uniform_1d)、ir_dict工作流(配合IrDictPolynomial和dict[Irrep, Array])执行等变多项式,以及…

npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax