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