torch~=1.11.0
torchvision~=0.12.0
bitorchinfo
matplotlib
numpy
sklearn
pytorch_lightning
torchmetrics

[dev]
build
flake8
mypy
pep8-naming
pytest
sphinx
twine
nbclient==0.5.13
nbsphinx==0.8.8
myst-nb
nbsphinx-link==1.3.0

[opt]
tensorboardX
tensorboard
wandb
fvbitcore
