numpy
ml_dtypes
tokenizers
jinja2==3.1.6
huggingface_hub
flatbuffers
gguf==0.19.0

[cpu]
jax==0.11.2

[dev]
pytest
jax==0.11.2
torch
transformers
safetensors

[hy-nomas]
torch>=2.5
transformers==4.57.1
pyyaml
scipy
imageio-ffmpeg
pillow

[tpu]
jax[tpu]==0.11.2
