feat(jax/array-api): se_atten_v2#4289

wanghan-iapcm merged 1 commit intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

njzjz:jax-se-atten-v2njzjz/deepmd-kit:jax-se-atten-v2Copy head branch name to clipboard