Han Wang

You may need to merge the devel to solve the issues in the UTs

please edit the title of the PR to make it informative.

the conflict should be resolved.

please try `"set_davg_zero": fase`

I was using a GPU machine, the jax was installed by ``` pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ```

I stopped trying to use tf, dp and jax in the same python code...

The methods in the linear_atomic_models should be rethink.

> @wanghan-iapcm Different 3D conformations of a molecule have exactly the same label, and each epoch takes one of all 3D conformations of this molecule. Because molecule may only has...

you may need to import deepmd defined ops. https://github.com/deepmodeling/deepmd-kit/blob/f8a0b3128e51ee523515cce26e520dde09a973f9/deepmd/train/trainer.py#L23-L24

> I just realized that we don't need to resolve it after #3496 is conducted. I agree