fix(jax): restore batched neighbor statistics - #5545

fix(jax): restore batched neighbor statistics#5545

wanghan-iapcm merged 1 commit into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

njzjz:fix-jax-update-sel-oomnjzjz/deepmd-kit:fix-jax-update-sel-oomCopy head branch name to clipboard