jax>=0.4.0
jaxlib>=0.4.0
equinox>=0.10.0
optax>=0.1.0
numpy
scipy

[dev]
pytest
pytest-cov
pytest-timeout
matplotlib
jupyter
plotly
mypy
ruff

[gpu]
jax[cuda12]>=0.4.0

[viz]
matplotlib>=3.5
plotly>=5.0
