Unpin jax/jaxlib/flax/cola (#503)

* Unpin cola

* Remove pins

* Remove pins

* Bump version

Author: thomaspinder

LICENSE renamed to LICENSE.txt

@@ -39,7 +39,7 @@
 __description__ = "Didactic Gaussian processes in JAX"
 __url__ = "https://github.com/JaxGaussianProcesses/GPJax"
 __contributors__ = "https://github.com/JaxGaussianProcesses/GPJax/graphs/contributors"
-__version__ = "0.9.5"
+__version__ = "0.10.0"
 
 __all__ = [
     "base",

gpjax/__init__.py

@@ -21,16 +21,16 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
 dependencies = [
-  "jax<0.4.28",
-  "jaxlib<0.4.28",
+  "jax>=0.5.0",
+  "jaxlib>=0.5.0",
   "optax>0.2.1",
   "jaxtyping>0.2.10",
   "tqdm>4.66.2",
   "tensorflow-probability>=0.24.0",
   "beartype>0.16.1",
-  "cola-ml==0.0.5",
-  "flax<0.10.0",
-  "numpy<2.0.0",
+  "cola-ml>=0.0.7",
+  "flax>=0.10.0",
+  "numpy>=2.0.0",
 ]
 
 [tool.hatch.envs.default]