wandb>=0.15.0
pyyaml>=6.0
numpy>=1.19.0

[all]
torch>=1.9.0
tensorflow>=2.6.0

[dev]
pytest>=7.0
pytest-cov>=4.0
black>=22.0
isort>=5.10
mypy>=1.0
pre-commit>=2.20

[tensorflow]
tensorflow>=2.6.0

[torch]
torch>=1.9.0
