huggingface_hub
h5py
jax
flax
flaxmodels@ https://github.com/matthias-wright/flaxmodels.git
