diff --git a/README.md b/README.md index 2ece7d4..a934f02 100644 --- 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. \ No newline at end of file +descent) to update the network parameters. diff --git a/docs/index.md b/docs/index.md index 92b3065..cf75ef6 100644 --- 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