numpy
scipy
requests

[docs]
sphinx
sphinx_rtd_theme
numpydoc

[hpo]
optuna

[test]
pytest
pytest-cov
flake8
pydocstyle
packaging
pre-commit

[train]
jax
optax
flowjax
equinox
