torch>=2.2.0
transformers>=4.36.0
accelerate>=0.25.0
numpy>=1.24.0
einops>=0.7.0
tqdm>=4.66.0
safetensors>=0.4.0
huggingface-hub>=0.20.0

[all]
pytest>=7.0
black>=23.0
flake8>=6.0
mypy>=1.0
isort>=5.12.0
sphinx>=5.0
sphinx-rtd-theme>=1.0
myst-parser>=0.18.0
triton>=2.2.0
flash-attn>=2.5.0
xformers>=0.0.23
deepspeed>=0.12.0
torchvision>=0.17.0
timm>=0.9.0
open-clip-torch>=2.24.0
vllm>=0.3.0
wandb>=0.16.0
tensorboard>=2.15.0

[dev]
pytest>=7.0
black>=23.0
flake8>=6.0
mypy>=1.0
isort>=5.12.0

[distributed]
deepspeed>=0.12.0

[docs]
sphinx>=5.0
sphinx-rtd-theme>=1.0
myst-parser>=0.18.0

[gpu]
triton>=2.2.0
flash-attn>=2.5.0
xformers>=0.0.23

[inference]
vllm>=0.3.0

[logging]
wandb>=0.16.0
tensorboard>=2.15.0

[multimodal]
torchvision>=0.17.0
timm>=0.9.0
open-clip-torch>=2.24.0
