torch>=1.11.0
pytorch-lightning>=2.0.0
jsonargparse[signatures]
datasets>=2.3.0
evaluate>=0.3.0
scikit-learn>=1.0.0

[timm]
timm>=0.9

[transformers]
sentence-transformers>=2.2.2

[wandb]
wandb
