fix jaxlib version

vitkl · web-flow · commit fb6715f8d206 · 2022-04-16T18:30:35.000+01:00
diff --git a/setup.cfg b/setup.cfg
@@ -19,6 +19,7 @@ install_requires =
     torch>=1.9.0
     pymc3>=3.8<3.10
     jax>=0.3,<=0.3.6
+    jaxlib>=0.1.65,<=0.3.5
     arviz==0.10.0
     numpy
     pandas
