accelerate<1.3,>=1.2.1
aiohttp<3.12,>=3.10
aiofiles<25,>=22.1.0
aioresponses<0.8,>=0.7.6
backoff<2.3,>=2.2.1
datasets<3.3,>=3.2.0
jsonlines
lm_eval[wandb]<0.5.0,>=0.4.5
mlflow<2.22.0,>=2.21.2
numpy<2.0,>=1.26.4
omegaconf<2.5,>=2.4.0dev3
packaging
pandas<3,>=2.0.3
peft<0.15,>=0.14.0
pexpect<4.9,>=4.8.0
pillow<11.2,>=11.1.0
protobuf>=5.29.0
pydantic<2.10,>=2.9.2
responses<0.26,>=0.25.0
skypilot<0.8,>=0.7.0
tensorboard<2.19,>=2.18.0
torch<2.6.0,>=2.5.0
torchdata<0.10.0,>=0.9.0
torchvision<0.21,>=0.20.0
tqdm
transformers<4.50,>=4.49.0
trl<0.16,>=0.15.0
typer
typing_extensions
wandb<0.20,>=0.19.3

[aws]
skypilot[aws]

[azure]
skypilot[azure]

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

[ci_cpu]
oumi[dev,docs,gcp]
vllm<0.8.0,>=0.7.3
llama-cpp-python<0.4,>=0.3.5

[ci_gpu]
oumi[dev,docs,gcp,gpu]
vllm<0.8.0,>=0.7.3
alpaca-eval<0.7,>=0.6.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.6
langdetect
immutabledict
nltk>=3.9.1
antlr4-python3-runtime==4.11
sympy>=1.12
sentencepiece>=0.1.98

[file_formats]
pdf2image<1.18,>=1.17.0
python-poppler<0.5,>=0.4.1

[gcp]
google-api-core>=2.19.0
google-api-python-client
google-auth>=2.30.0
google-cloud-core>=2.4.1
google-cloud-storage>=2.17.0
skypilot[gcp]

[gpu]
liger-kernel<0.6,>=0.5.0
nvidia-ml-py<12.561,>=12.560.30
bitsandbytes<0.46,>=0.45.0

[lambda]
skypilot[lambda]

[runpod]
skypilot[runpod]
