torch~=1.12.0
torchvision~=0.13.0
matplotlib
numpy

[dev]
black
build
flake8
flake8-docstrings
mypy~=0.920
myst-nb
nbclient==0.5.13
nbsphinx-link==1.3.0
nbsphinx==0.8.8
pep8-naming
pre-commit
pytest
pytest-cov
sphinx
twine

[opt]
fvbitcore
pytorch_lightning~=1.6.0
sklearn
wandb~=0.12.0
