datasets>=4.0.0
flax>=0.12.0
optax>=0.2.5
pillow>=11.3.0
rich>=14.1.0
safetensors>=0.6.2
transformers>=4.56.1
typer>=0.17.4
peft
hf_transfer
cloudpathlib>=0.23.0
jax>=0.8

[aws]
cloudpathlib[s3]

[azure]
cloudpathlib[azure]

[dev]
mkdocs
mkdocs-material
pytest
pytest-forked
torch
ty
cloudpathlib[s3]

[gcp]
cloudpathlib[gs]

[gpu]
jax[cuda12]>=0.7.2

[tinker]
tinker>=0.2.1
fastapi[standard]
sqlmodel
sqlalchemy[asyncio]
aiosqlite
asyncpg
psycopg2-binary

[tpu]
jax[tpu]>=0.7.2
