feat(jax): energy, dos, dipole, polar, property atomic model & model#4384

wanghan-iapcm merged 3 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

njzjz:jax-model3njzjz/deepmd-kit:jax-model3Copy head branch name to clipboard