Update pyproject.toml

Co-authored-by: Matthew R. Becker <beckermr@users.noreply.github.com>

Author: EiffL

pyproject.toml

@@ -16,7 +16,7 @@ readme = "README.md"
 dependencies = [
     "numpy >=1.18.0",
     "galsim >=2.3.0",
-    "jax >=0.4.0",
+    "jax >=0.8.0",
     "astropy >=2.0",
     "tfp-nightly",
     "quadax",