submitit>=1.4.0

[all]
huggingface_hub
wandb
mlflow
torch
torchvision
torchaudio
tensorboard
matplotlib
seaborn

[dev]
pytest>=6.0
pytest-cov
pytest-mock
black
flake8
mypy
isort

[huggingface]
huggingface_hub

[metrics]
tensorboard
matplotlib
seaborn

[torch]
torch
torchvision
torchaudio

[tracking]
wandb
mlflow
