torch>=1.4
torchvision
ray[default,tune]<1.9,>=1.6
timm
accelerate>=0.5.0
smart_open<=5.2.1,>=5.1
rich
loguru>=0.5
torchmetrics>=0.5.0

[dev]
lightning-flash[image,text]==0.5.1
codecarbon>=1.2.0
comet_ml
wandb

[test]
pytest
coverage
pytest-sugar
