torch==2.9.0
torchvision==0.24.0
torchaudio==2.9.0
datasets==4.0.0
transformers==4.56.1
trl==0.23.0
wandb==0.21.0
peft==0.17.0
ipykernel==6.30.1
ipywidgets==8.1.7
fsspec==2025.3.0
huggingface_hub==0.34.4
tiktoken==0.12.0
boto3==1.42.78

[cuda]
bitsandbytes==0.46.1
deepspeed==0.17.4
mpi4py==4.1.0

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