jax
jaxlib
numpy
scikit-learn
scipy
tqdm
torch

[dev]
black
flake8
flake8-bugbear
flake8-blind-except
flake8-builtins
flake8-docstrings
flake8-mutable
hypothesis
isort
mypy
pep8-naming
pytest
