feat(jax): freeze models with hessian output#5608

njzjz wants to merge 5 commits into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

njzjz:feat/jax-hessian-freezenjzjz/deepmd-kit:feat/jax-hessian-freezeCopy head branch name to clipboard