numpy
scipy
scikit-learn
torch
tqdm

[dev]
pytest
pytest-lazy-fixture
tensorflow
skorch
keras
sphinx
sphinx_rtd_theme
black

[test]
pytest
pytest-lazy-fixture
tensorflow
skorch
keras
