Bump jax dependency up to 0.5.3

Author: JanLuca

pyproject.toml

@@ -14,7 +14,7 @@ requires-python = ">=3.10"
 dependencies = [
     "numpy >= 1.21.2",
     "scipy >= 1.7.1",
-    "jax[cpu]>=0.5.1",
+    "jax[cpu]>=0.5.3",
     "tensornetwork >= 0.4.5",
     "h5py >= 3.6.0",
     "tqdm[notebook] >= 4.64.1",