jax>=0.4
jaxlib>=0.4
equinox>=0.11
optax>=0.2
multipledispatch
numpy
optuna>=3
scikit-learn
tqdm

[dev]
pytest
pytest-cov
hypothesis
black
isort
pre-commit
mkdocs
mkdocs-material
mkautodoc
pyprojroot
