torch>=2.0.0
numpy>=1.21.0
tqdm>=4.64.0

[dev]
pytest>=7.0.0
black>=22.0.0
flake8>=5.0.0
jupyter>=1.0.0
build>=0.10.0
twine>=4.0.0

[docs]
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0

[full]
wandb>=0.15.0
matplotlib>=3.5.0
scipy>=1.9.0
scikit-learn>=1.1.0
