torch
torchvision
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
pytest
sphinx
twine
pre-commit

[opt]
bitorch_inference_engine
wandb
fvbitcore
sklearn
pytorch_lightning
torchmetrics
