accelerate>=1.4.0
addict>=2.4.0
altair[all]>=5.5.0
captum>=0.7.0
colorama>=0.4.6
datasets<=3.2.0
einops>=0.7.0
evaluate>=0.4.3
huggingface-hub>=0.29.0
ipykernel>=6.29.5
ipywidgets>=8
aiohttp>=3.9.0
jax>=0.5.2
loguru>=0.7.0
mambapy>=1.2.0
mcp>=1.3.0
modelscope[framework]>=1.23.2
oss2>=2.18.0
nbstripout>=0.8.1
ninja>1
numpy<2
openpyxl>=3.1.5
pandas>=2.2.3
peft>=0.14.0
pydantic>=2.10.6
python-dotenv>=1.0.0
pyyaml>=6.0
rich>=13.7.0
scikit-learn>=1.4.0
scipy>=1.15.2
sentencepiece>=0.2.0
seqeval>=1.2.2
starlette<0.48.0,>=0.40.0
tensorboardx>=2.6.2.2
tokenizers>=0.21.0
torch<=2.7,>=2.4.0
tqdm>=4.67.1
transformers<4.56.0,>=4.49.0
umap-learn>=0.5.7
uvicorn>=0.24.0
wandb>=0.19.8
websockets>=12.0
wheel>=0.45.0

[base]
asyncio>=3.4.3
ruff>=0.1.12
flake8>=7.0.0
ipywidgets>=8
isort>=6.0.1
jupyter>=1.1.1
marimo>=0.16.3
mcp>=1.3.0
mypy>=1.15.0
pandas-stubs>=2.3.0
pydantic>=2.10.6
types-transformers>=0.1.0
pytest>=8.3.5
pytest-cov>=6.0.0
pytest-progress>=0.1.0

[cpu]
torch<2.7,>=2.4.0

[cuda121]
torch<2.7,>=2.2.0

[cuda124]
torch<2.7,>=2.4.0

[cuda126]
torch<=2.7,>=2.6.0

[cuda128]
torch<=2.7,>=2.6.0

[dev]
ruff>=0.1.12
flake8>=7.0.0
pre-commit>=3.6.0
ipywidgets>=8
jupyter>=1.1.1
marimo>=0.16.3
mypy>=1.15.0
pandas-stubs>=2.3.0
pydantic>=2.10.6
pytest>=8.3.5
pytest-asyncio>=0.21.1
pytest-cov>=6.0.0
pytest-progress>=0.1.0
pytest-timeout>=2.3.1

[docs]
mkdocs-jupyter>=0.24.0
mkdocs-material>=9.6.1
mkdocstrings-python>=1.16.10
click<8.2.2

[mamba]
causal_conv1d==1.6.0
mamba-ssm==2.3.0
torch<=2.7,>=2.6.0

[mcp]
mcp>=1.3.0
asyncio>=3.4.3

[notebook]
ipywidgets>=8
jupyter>=1.1.1
marimo>=0.16.3

[rocm]
torch<2.7,>=2.5.0

[test]
pytest>=8.3.5
pytest-asyncio>=0.21.1
pytest-cov>=6.0.0
pytest-progress>=0.1.0
pytest-timeout>=2.3.1
