jax[cpu]
numpy
pandas
matplotlib
jaxley>=0.2.0
diffrax>=0.6
equinox>=0.11.8

[dev]
ruff
jupyter
ipykernel
mkdocs
mkdocs-material
markdown-include
mkdocs-redirects
mkdocstrings[python]>=0.18
neuron
pytest
pyright
watermark
tqdm
