feat(jax/array-api): se_e2_r#4257

wanghan-iapcm merged 2 commits into

deepmodeling:develdeepmodeling/deepmd-kit:develfrom

njzjz:jax-se-e2-rnjzjz/deepmd-kit:jax-se-e2-rCopy head branch name to clipboard