torch==2.7.1
torchvision==0.22.1
torchaudio==2.7.1
datasets==4.0.0
transformers==4.56.1
trl==0.23.0
wandb==0.21.0
bitsandbytes==0.46.1
peft==0.17.0
deepspeed==0.17.4
ipykernel==6.30.1
ipywidgets==8.1.7
fsspec==2025.3.0
huggingface_hub==0.34.4
mpi4py==4.1.0

[dev]
pytest>=7.0.0
black>=22.0.0
flake8>=4.0.0
mypy>=0.950
