huggingface_hub

[gpu]
jax[cuda12]

[metal]
jax==0.4.38
jaxlib==0.4.38
jax-metal

[test]
pytest
pydantic

[train]
numpy
jax
jaxlib
flax>=0.10.2
optax
sentencepiece
