fix orbax-checkpoint version for consistency

Author: kabylda

pyproject.toml

@@ -13,6 +13,7 @@ ase = "*"
 h5py = "*"
 jax = "==0.5.3"
 jaxlib = "==0.5.3"
+orbax-checkpoint = "==0.11.13"
 setuptools = "<81"
 mlff = { git = "https://github.com/kabylda/mlff.git", branch = "v1.0-tfds-pme", optional = false }
 jraph = { git = "https://github.com/kabylda/jraph.git", branch = "master", optional = false }