jax
equinox
einops
huggingface-hub
