torch~=2.2.2
numpy~=1.26.0
lightning~=2.4.0
wandb~=0.16.6
pyyaml~=6.0.1
torchvision~=0.17.2
datasets~=2.19.0
transformers~=4.40.1
python-dotenv~=1.0.1
requests~=2.32.3
uvicorn~=0.30.5
fastapi~=0.112.0
huggingface-hub~=0.23.4
scikit-learn~=1.5.1
strenum~=0.4.15
