numpy>=1.26.4
huggingface_hub[cli]>=0.23.0
safetensors>=0.5.3

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

[torch]
torch>=2.0.0

[xreg]
jax[cuda]
scikit-learn
