numpy
scipy
jax
optax
flax
brax

[dev]
pathos
tqdm
jupyter
ipython
pytest
flake8
flake8-print
flake8-bugbear
mypy
pytest-cov
pytest-instafail
pytest-xdist
pylint
black
reorder-python-imports
autoflake
pre-commit
pydocstring-coverage
bumpversion
ipython
jupyter
gprof2dot
bandit
check-python-versions
line-profiler

[docs]
sphinx
sphinxcontrib-napoleon
sphinx_rtd_theme
