docs: Update JAX backend documentation on training support#5482

wanghan-iapcm merged 1 commit into

deepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

njzjz:docs-jax-trainingnjzjz/deepmd-kit:docs-jax-trainingCopy head branch name to clipboard