feat(jax): atomic virial#4290

wanghan-iapcm merged 6 commits into

deepmodeling:develdeepmodeling/deepmd-kit:develfrom

njzjz:jax-atomic-virialnjzjz/deepmd-kit:jax-atomic-virialCopy head branch name to clipboard