fix(jax): add hessian energy loss#5607

njzjz wants to merge 2 commits into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

njzjz:fix/dpmodel-energy-hessian-lossnjzjz/deepmd-kit:fix/dpmodel-energy-hessian-lossCopy head branch name to clipboard