accelerate<2.0,>=1.10.0
aiohttp<3.16,>=3.12
aiofiles<26,>=24.1.0
aioresponses<0.8,>=0.7
backoff<2.3,>=2.2.1
click<8.4.0
datasets<4.8.6,>=3.2
greenlet
hdrhistogram<0.11,>=0.10
httpx<1.0,>=0.27
jsonlines
jsonschema<5.0,>=4.0
lm_eval[wandb]<0.5.0,>=0.4
mlflow>=3.1
numpy<2.4,>=1.26
omegaconf==2.4.0.dev4
packaging
pandas<3,>=2.3
peft<0.20,>=0.17
pexpect<4.10,>=4.9
posthog<8.0,>=7.0
pillow<12.3,>=11.3
protobuf>=6.32
pycares<6.0.0
pydantic<2.14,>=2.11
requests<3.0,>=2.31
responses<0.27,>=0.25
safetensors<0.8,>=0.6
skypilot<0.13,>=0.11.1
tensorboard<2.21,>=2.20
tiktoken<1.0,>=0.7
torch<2.11.0,>=2.6
torchao<0.17,>=0.16
torchvision<0.26,>=0.21
tqdm
transformers<5.8,>=4.57
trl<1.4,>=0.24
typer<0.25.2
typing_extensions
uvicorn<0.45.0
wandb<0.26,>=0.21

[aws]
skypilot[aws]

[azure]
skypilot[azure]

[bitnet]
onebitllms>=0.0.3

[ci_cpu]
oumi[dev,docs,file_formats,gcp,mcp,synthesis,torchdata,tune]
vllm<0.21,>=0.14
boto3
llama-cpp-python<0.4,>=0.3

[ci_gpu]
oumi[deepspeed,dev,docs,file_formats,gcp,gpu,synthesis,torchdata,tune]
boto3

[deepspeed]
deepspeed<0.19.0,>=0.17.0

[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
sphinxcontrib-bibtex
sphinxcontrib-mermaid
sphinxcontrib-typer

[evaluation]
langdetect
immutabledict
nltk>=3.9
antlr4-python3-runtime==4.13.2
sympy>=1.14
sentencepiece>=0.2.0

[file_formats]
pdf2image<1.18,>=1.17

[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.9,>=0.6
nvidia-ml-py<13.596,>=13.580
bitsandbytes<0.50,>=0.47
vllm<0.21,>=0.14
kernels<0.14,>=0.11

[gpu:python_version >= "3.10"]
verl<0.8,>=0.5

[kubernetes]
skypilot[kubernetes]

[lambda]
skypilot[lambda]

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

[mcp]
fastmcp<4,>=3.0.0
httpx<1,>=0.28.1
mcp<2,>=1.25.0

[nebius]
skypilot[nebius]

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

[runpod]
skypilot[runpod]

[synthesis]
python-docx>=1.1.0
openpyxl>=3.1.0

[synthesis:python_version >= "3.10"]
pdftext>=0.6.3

[torchdata]
torchdata<0.10.0,>=0.9

[tune]
optuna<5.0,>=4.0.0
