torch==2.1.0
torchvision==0.16.0
torchaudio==2.1.0
lightning==2.1.0
pytorch-lightning==2.1.0
transformers==4.35.0
numpy<2.0
scipy==1.13.1
scikit-learn==1.6.1
matplotlib==3.9.4
seaborn==0.13.2
wandb==0.25.1
python-dotenv==1.1.0
psutil
requests==2.28.2
urllib3<2.0.0
chardet==4.0.0
charset-normalizer==3.0.1
datetime==5.5

[cuda]
deepspeed==0.14.5
bitsandbytes==0.42.0
nvidia-ml-py==13.580.82

[dev]
pre-commit
debugpy

[tensorboard]
tensorboard
