torch>=2.2.0
torchvision>=0.17.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
packaging>=20.0

[all]
parallel-llm[datasets,dev,distributed,docs,gpu,inference,logging,multimodal]

[cpu]

[datasets]
datasets>=2.0.0
pyarrow>=10.0.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]
Pillow>=9.0.0
requests>=2.25.0
timm>=0.9.0
open-clip-torch>=2.24.0
