numpy>=1.26.4
huggingface_hub>=0.28.0
safetensors>=0.5.3

[flax]
flax
optax
einshape
orbax-checkpoint
jaxtyping
jax[cuda]

[mlx]
mlx>=0.32
safetensors>=0.4.0
huggingface-hub>=0.24.0

[torch]
torch>=2.0.0

[xreg]
jax[cuda]
scikit-learn
