accelerate<2.0,>=1.10.0
aiohttp<3.16,>=3.12
aiofiles<25,>=24.1.0
aioresponses<0.8,>=0.7
backoff<2.3,>=2.2.1
click<8.3.0
datasets<3.3,>=3.2
hdrhistogram<0.11,>=0.10
jsonlines
lm_eval[wandb]<0.5.0,>=0.4
mlflow>=3.1
numpy<2.4,>=1.26
omegaconf==2.4.0.dev3
packaging
pandas<3,>=2.3
peft<0.18,>=0.17
pexpect<4.10,>=4.9
pillow<11.4,>=11.3
protobuf>=6.32
pydantic<2.12,>=2.11
responses<0.26,>=0.25
safetensors<0.7,>=0.6
skypilot<0.11,>=0.10.2
tensorboard<2.21,>=2.20
torch<2.9.0,>=2.6
torchao<0.13,>=0.12
torchdata<0.10.0,>=0.9
torchvision<0.24,>=0.21
tqdm
transformers<4.58,>=4.57
trl<0.26,>=0.25
typer
typing_extensions
uvicorn<0.36.0
wandb<0.22,>=0.21

[aws]
skypilot[aws]

[azure]
skypilot[azure]

[bitnet]
onebitllms>=0.0.3

[cambrian]
timm==0.9.16
open_clip_torch
diffusers[torch]
einops==0.6.1

[ci_cpu]
oumi[dev,docs,gcp,synthesis,tune]
vllm<0.11,>=0.10
llama-cpp-python<0.4,>=0.3

[ci_gpu]
oumi[dev,docs,gcp,gpu,synthesis,tune]
alpaca-eval<0.7,>=0.6

[dev]
ipykernel
ipywidgets
jupyterlab
nbconvert
nbformat
pre-commit
pyright
pytest-asyncio
pytest-cov
pytest-timeout
pytest
responses
ruff
torchfix

[docs]
myst_parser
myst-nb
sphinx-book-theme
sphinx-copybutton
sphinx-design
sphinx-rtd-theme>=3.0.1
sphinx
sphinxcontrib-bibtex
sphinxcontrib-mermaid
sphinxcontrib-typer

[evaluation]
alpaca-eval<0.7,>=0.6
langdetect
immutabledict
nltk>=3.9
antlr4-python3-runtime==4.11
sympy>=1.14
sentencepiece>=0.2.0

[file_formats]
pdf2image<1.18,>=1.17
python-poppler<0.5,>=0.4

[flash_attn]
flash-attn>=2.0.0

[gcp]
google-api-core>=2.25
google-api-python-client
google-auth>=2.40
google-cloud-core>=2.4
google-cloud-storage>=3.3
skypilot[gcp]

[gpu]
liger-kernel<0.7,>=0.6
nvidia-ml-py<13.581,>=13.580
bitsandbytes<0.48,>=0.47
verl<0.6,>=0.5
vllm<0.11,>=0.10
deepspeed<0.18.0,>=0.17.0

[lambda]
skypilot[lambda]

[llama_cpp]
llama-cpp-python<0.4,>=0.3.5

[quantization]
autoawq<0.3,>=0.2
bitsandbytes<0.48,>=0.47

[runpod]
skypilot[runpod]

[synthesis]
pdftext>=0.6.3

[tune]
optuna<5.0,>=4.0.0
