cuequivariance-jax
bởi nvidia
Thực thi các đa thức tương đương trong JAX bằng segmented_polynomial (naive/uniform_1d), quy trình ir_dict với IrDictPolynomial và dict[Irrep, Array], và…
npx skills add https://github.com/nvidia/cuequivariance --skill cuequivariance-jax