chore: uv lock --upgrade

Updated jax v0.4.30, v0.4.38 -> v0.4.30, v0.5.3
Updated jaxlib v0.4.30, v0.4.38 -> v0.4.30, v0.5.3
Updated orbax-checkpoint v0.6.4, v0.11.5 -> v0.6.4, v0.11.10
Updated pyparsing v3.2.1 -> v3.2.2
Updated setuptools v77.0.3 -> v78.0.1

Author: rg936672

requirements-doc.txt

@@ -35,9 +35,9 @@ imagesize==1.4.1
 importlib-metadata==8.6.1 ; python_full_version < '3.10'
 importlib-resources==6.5.2
 jax==0.4.30 ; python_full_version < '3.10'
-jax==0.4.38 ; python_full_version >= '3.10'
+jax==0.5.3 ; python_full_version >= '3.10'
 jaxlib==0.4.30 ; python_full_version < '3.10'
-jaxlib==0.4.38 ; python_full_version >= '3.10'
+jaxlib==0.5.3 ; python_full_version >= '3.10'
 jaxopt==0.8.3
 jaxtyping==0.2.36 ; python_full_version < '3.10'
 jaxtyping==0.3.0 ; python_full_version >= '3.10'
@@ -59,7 +59,7 @@ numpy==2.1.3 ; python_full_version >= '3.10'
 opt-einsum==3.4.0
 optax==0.2.4
 orbax-checkpoint==0.6.4 ; python_full_version < '3.10'
-orbax-checkpoint==0.11.5 ; python_full_version >= '3.10'
+orbax-checkpoint==0.11.10 ; python_full_version >= '3.10'
 packaging==24.2
 platformdirs==4.3.7
 protobuf==6.30.1
@@ -77,7 +77,7 @@ ruamel-yaml-clib==0.2.12 ; python_full_version < '3.13' and platform_python_impl
 scikit-learn==1.6.1
 scipy==1.13.1 ; python_full_version < '3.10'
 scipy==1.15.2 ; python_full_version >= '3.10'
-setuptools==77.0.3 ; python_full_version >= '3.12'
+setuptools==78.0.1 ; python_full_version >= '3.12'
 simplejson==3.20.1 ; python_full_version >= '3.10'
 six==1.17.0
 snowballstemmer==2.2.0