absl-py>=2.5
distrax>=0.1.9
dm-tree>=0.1.10
equinox==0.13.6
immutabledict>=4.3
jax==0.11.0
jaxlib==0.11.0
jaxtyping==0.3.9
networkx>=3.5
numpy>=2.3
optax>=0.2.8
packaging>=26
typing-extensions>=4.15

[notebooks]
jupyterlab>=4.4
matplotlib>=3.10
