jax
equinox
einops
diffrax
huggingface-hub
