flax==0.10.4
safetensors
pydantic
huggingface-hub
tokenizerz==0.0.2a5

[:platform_system == "Darwin"]
jax-metal==0.1.1
jaxlib==0.5.0
jax==0.5.0

[gpu]
jax[cuda12]

[tpu]
jax[tpu]
