torch>=2.0.0
torchvision>=0.15.0
timm>=0.9.0
pyyaml>=6.0
numpy>=1.21.0

[advanced]
flash-attn>=2.0.0
bitsandbytes>=0.41.0

[all]
transformers>=4.30.0
datasets>=2.12.0
tokenizers>=0.13.0
deepspeed>=0.9.0
accelerate>=0.20.0
wandb>=0.15.0
tensorboard>=2.13.0

[dev]
pytest>=7.3.0
pytest-cov>=4.1.0
black>=23.3.0
flake8>=6.0.0
mypy>=1.3.0
isort>=5.12.0

[distributed]
deepspeed>=0.9.0
accelerate>=0.20.0

[llm]
transformers>=4.30.0
datasets>=2.12.0
tokenizers>=0.13.0

[tracking]
wandb>=0.15.0
tensorboard>=2.13.0
