Update dependencies description

thebuckleylab · Jun 12, 2024 · c2a20d3 · c2a20d3
1 parent 71e14c2
commit c2a20d3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -18,9 +18,9 @@ useful JAX transforms such as `vmap` and `jit`.
 pip install jpc
 ```
 
-Requires Python 3.8+, JAX 0.4.13+, [Equinox](https://github.com/patrick-kidger/equinox) 
-0.10.4+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.3.1+, and 
-[Jaxtyping](https://github.com/patrick-kidger/jaxtyping) 0.2.19+.
+Requires Python 3.9+, JAX 0.4.23+, [Equinox](https://github.com/patrick-kidger/equinox) 
+0.11.2+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.5.1+, and 
+[Jaxtyping](https://github.com/patrick-kidger/jaxtyping) 0.2.24+.
 
 ## Documentation
 Available at X.
@@ -70,4 +70,4 @@ pc_param_grads = jpc.compute_pc_param_grads(
 )
 ```
 The gradients can then be fed to your favourite optimiser (e.g. gradient
-descent) to update the network parameters.
+descent) to update the network parameters.
diff --git a/docs/index.md b/docs/index.md
@@ -11,17 +11,15 @@ activity (inference) dynamics.
 JPC provides a simple but flexible API for research of PCNs compatible with
 useful JAX transforms such as `vmap` and `jit`.
 
-
 ## Installation
 
 ```
 pip install jpc
 ```
 
-Requires Python 3.8+, JAX 0.4.13+, [Equinox](https://github.com/patrick-kidger/equinox) 
-0.10.4+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.3.1+, and 
-[Jaxtyping](https://github.com/patrick-kidger/jaxtyping) 0.2.19+.
-
+Requires Python 3.9+, JAX 0.4.23+, [Equinox](https://github.com/patrick-kidger/equinox) 
+0.11.2+, [Diffrax](https://github.com/patrick-kidger/diffrax) 0.5.1+, and 
+[Jaxtyping](https://github.com/patrick-kidger/jaxtyping) 0.2.24+.
 
 ## Quick example
 
@@ -70,5 +68,4 @@ pc_param_grads = jpc.compute_pc_param_grads(
 The gradients can then be fed to your favourite optimiser (e.g. gradient
 descent) to update the network parameters.
 
-
 ## Citation