diff --git a/var/spack/repos/builtin/packages/py-flax/package.py b/var/spack/repos/builtin/packages/py-flax/package.py
index f1bbc9678ed..108ddfcc654 100644
--- a/var/spack/repos/builtin/packages/py-flax/package.py
+++ b/var/spack/repos/builtin/packages/py-flax/package.py
@@ -14,6 +14,7 @@ class PyFlax(PythonPackage):
 
     license("Apache-2.0")
 
+    version("0.8.5", sha256="4a9cb7950ece54b0addaa73d77eba24e46138dbe783d01987be79d20ccb2b09b")
     version("0.8.1", sha256="ce3d99e9b4c0d2e4d9fc28bc56cced8ba953adfd695aabd24f096b4c8a7e2f92")
     version("0.7.3", sha256="e9dbc7eb6c80d31277f97b626c07978d2a84f1bb635cf05957a02a3a496493e6")
 
@@ -23,8 +24,11 @@ class PyFlax(PythonPackage):
 
     with default_args(type=("build", "run")):
         depends_on("python@3.9:", when="@0.8:")
+        depends_on("py-numpy@1.26.0:", when="@0.8: ^python@3.12:")
+        depends_on("py-numpy@1.23.2:", when="@0.8: ^python@3.11:")
         depends_on("py-numpy@1.22:", when="@0.8:")
         depends_on("py-numpy@1.12:")
+        depends_on("py-jax@0.4.27:", when="@0.8.5:")
         depends_on("py-jax@0.4.19:", when="@0.8:")
         depends_on("py-jax@0.4.2:")
         depends_on("py-msgpack")