click<=8.2.1
flax>=0.12.2
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

[cpu]
jax>=0.4.23
torch>=2.9.1
torchax>=0.0.11
tokenizers>=0.22.2
transformers>=5.1.0

[cuda12]
jax[cuda12]>=0.4.23
torch>=2.9.1
torchax>=0.0.11
tokenizers>=0.22.2
transformers>=5.1.0

[cuda13]
jax[cuda13]>=0.4.23
torch>=2.9.1
torchax>=0.0.11
tokenizers>=0.22.2
transformers>=5.1.0

[dev]
coverage>=7.9.1
coveralls>=4.0.1
pytest>=8.4.1
pytest-cov>=6.2.1
ruff>=0.12.1
pre-commit>=4.2.0
mypy>=1.16.1

[docs]
mkdocs>=1.6.1
mkdocs-material>=9.6.20
mkdocstrings[python]>=0.30.0
pymdown-extensions>=10.15
mkdocs-gen-files>=0.5.0
mkdocs-literate-nav>=0.6.0
mkdocs-section-index>=0.3.0

[fever]
wikipedia>=1.4.0

[huggingface]
tokenizers>=0.22.2
transformers>=5.1.0

[tpu]
jax[tpu]>=0.4.23
torch>=2.9.1
torchax>=0.0.11
tokenizers>=0.22.2
transformers>=5.1.0

[web]
fastapi>=0.100.0
uvicorn[standard]>=0.23.0
jinja2>=3.1.0
aiofiles>=23.0.0
sse-starlette>=1.6.0
pyyaml>=6.0.0
python-multipart>=0.0.22
bcrypt>=5.0.0
itsdangerous>=2.2.0
watchdog>=4.0.0
