numpy>=1.15
torch>=1.8
scipy>=1.2

[dev]
torch==1.8.1
torchvision>=0.9.1
tqdm>=4.40
requests>=2.25.1
black
pytest
flake8
sphinx
sphinx-autodoc-typehints
mypy>=0.760
isort
torchcsprng
hypothesis
tensorboard
datasets
transformers
scikit-learn
pytorch-lightning
lightning-bolts
jsonargparse[signatures]>=3.19.3
