jax<0.8,>=0.7
flax<0.13,>=0.12
optax<0.3,>=0.2
transformers<4.40,>=4.38
safetensors
numpy
scikit-learn
tqdm

[cuda12]
jax[cuda12]<0.8,>=0.7

[demo]
gradio<6,>=4

[dev]
pytest

[docs]
sphinx>=7
myst-parser
furo

[finetune]
torch
accelerate
sentencepiece
peft<0.14,>=0.10

[gemma]
torch

[leolm]
torch
accelerate
sentencepiece
