Restore BaseModel.update_sel once the JAX data path avoids OOM.

[github-actions]

Line: 202

deepmd-kit/deepmd/jax/entrypoints/train.py

Lines 199 to 203 in 2087416

	"Skip neighbor statistics update for JAX training; "
	"BaseModel.update_sel currently needs more memory than expected."
	)
	# TODO: Restore BaseModel.update_sel once the JAX data path avoids OOM.
	return jdata.copy()