click>=8.2.1
cloudpickle>=3.1.1
flax>=0.12.2
flatbuffers>=25.12.0
jsonlines>=4.0.0
loguru>=0.7.3
numpy>=2.4.1
omegaconf>=2.3.0
orbax>=0.1.9
pydantic>=2.12.5
python-dotenv>=1.2.1
rich>=13.5.2
seaborn>=0.13.2
tiktoken>=0.12.0
torchax>=0.0.11
wandb>=0.24.1
datasets>=4.5.0
huggingface-hub>=1.4.1
flatbuffers>=25.12.19
pyarrow>=23.0.1
chdb>=4.3.0
deltalake>=1.6.3
typer>=0.24.1
nicegui<4,>=3.16.0
pygwalker>=0.5.0.1
pandas>=2
comet-ml>=3.58.6
bottle>=0.13.4
cheroot>=11.1.2

[all]
tokenizers>=0.22.2
transformers>=5.1.0
accelerate>=1.13.0
coverage>=7.9.1
coveralls>=4.0.1
pytest>=8.4.1
playwright>=1.55.0
pytest-cov>=6.2.1
ruff>=0.12.1
pre-commit>=4.2.0
mypy>=1.16.1
ipdb>=0.13.13
xprof~=2.21.5
zensical<0.1,>=0.0.60
mkdocstrings-python>=2.0.3
pymdown-extensions>=10.15
wikipedia>=1.4.0

[cpu]
jax>=0.4.23
torch>=2.9.1
torchax>=0.0.11

[cuda12]
jax[cuda12]>=0.4.23
torch>=2.9.1
torchax>=0.0.11

[cuda13]
jax[cuda13]>=0.4.23
torch>=2.9.1
torchax>=0.0.11

[dev]
coverage>=7.9.1
coveralls>=4.0.1
pytest>=8.4.1
playwright>=1.55.0
pytest-cov>=6.2.1
ruff>=0.12.1
pre-commit>=4.2.0
mypy>=1.16.1
ipdb>=0.13.13
xprof~=2.21.5

[docs]
zensical<0.1,>=0.0.60
mkdocstrings-python>=2.0.3
pymdown-extensions>=10.15

[fever]
wikipedia>=1.4.0

[huggingface]
tokenizers>=0.22.2
transformers>=5.1.0
accelerate>=1.13.0

[tpu]
jax[tpu]>=0.4.23
torch>=2.9.1
torchax>=0.0.11
