use last JAX release (#198)

ilabcode · Jun 12, 2024 · e0a2597 · e0a2597
1 parent 830c123
commit e0a2597
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -5,8  5,8 @@ dependencies:
   - conda-forge::pymc
   - graphviz
   - pip:
-    - jax==0.4.26
-    - jaxlib==0.4.26
     - jax==0.4.29
     - jaxlib==0.4.29
     - numpy==1.26.4
     - matplotlib>=3.0.2
     - seaborn>=0.9.0