jax
jaxlib
keras-core
keras-nlp
huggingface_hub