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