Update oryx version and add other requirements.

PiperOrigin-RevId: 592231745
jax-ml · Dec 19, 2023 · bac653e · bac653e
1 parent f526dd3
commit bac653e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,12 @@ keywords = []
 dependencies = [
   "jax>=0.4.6",
   "tensorflow-probability[jax]>=0.19.0",
-  "oryx>=0.2.5",
+  "oryx@git+https://github.com/jax-ml/oryx",
+  "arviz",
+  "optax",
+  "blackjax",
+  "numpyro",
+  "jaxopt",
 ]
 
 # `version` is automatically set by flit to use `bayeux.__version__`