aiohttp
click
datasets<=5.1.0,>=4.0.0
httpx
huggingface-hub
loguru<=0.7.3,>=0.7.2
numpy<=2.5.4,>=2.0.0
openai>=2.0.0
protobuf
psutil
pydantic>=2.0.0
pydantic-settings>=2.0.0
rich
safetensors
torch<=2.14.1,>=2.9.0
torchaudio
torchvision
tqdm<=4.70.1,>=4.66.3
transformers<=5.19.0,>=5.0.0
typer>=0.12.0
hs-connectors==0.9.0

[dev]
build>=1.6.1
setuptools>=84.0.0
setuptools-git-versioning<3,>=2.0
mike~=2.2.0
mkdocs~=1.6.1
mkdocs-material[imaging]~=9.7.7
markdown~=3.10.2
pymdown-extensions~=12.1.0
mkdocs-section-index~=0.3.12
mkdocs-minify-plugin~=0.8.0
mkdocs-api-autonav~=0.4.0
mkdocs-awesome-nav~=3.3
mkdocstrings-python~=2.0.9
mkdocs-gen-files~=0.6.1
mkdocs-nav-weight~=0.3.0
mkdocs-redirects~=1.2.3
pygments<2.22.0
sphinx~=8.1.3
pytest~=9.1.1
pytest-asyncio~=1.4.0
pytest-cov~=7.1.0
pytest-mock~=3.16.0
pytest-rerunfailures~=16.7
scipy
plotly
mypy~=2.1.0
ruff~=0.15.20
pre-commit~=4.6
mdformat~=1.0.0
mdformat-footnote~=0.1.3
mdformat-frontmatter~=2.1.2
mdformat-gfm~=1.0.0
mdformat-admon~=2.1.1
mdformat-gfm-alerts~=2.0.0
types-click~=7.1.8
types-PyYAML~=6.0.12
types-requests~=2.33.0
types-toml
types-tqdm
mkdocs-linkcheck~=1.0.6

[mooncake-cuda]
mooncake-transfer-engine>=0.3.12

[mooncake-cuda13]
mooncake-transfer-engine-cuda13>=0.3.13.post1
