Using KerasHub for easy end-to-end machine learning workflows with Hugging Face- Google Developers Blog
Jun 24, 2025Yufeng Guo, Divyashree Sreepathihalli, Monica Song

Summary
Introduces KerasHub's capability to load and utilize model checkpoints from Hugging Face Hub across different ML frameworks (JAX, PyTorch, TensorFlow). Details the integration with SafeTensors format and provides examples for loading Mistral, Llama 3.1, and Gemma models using specific KerasHub model classes (MistralCausalLM, Llama3CausalLM, Gemma3CausalLM). Highlights the flexibility of mixing model architectures with weights from various sources.
More from the JAX Model Evaluation Metrics trail
- Accelerating Spatio-Temporal Attention for Video Diffusion on TPUs- Google Developers Blog
- Reproducing OLMo 3 7B Pre-training in MaxText: case study of large scale training on TPUs- Google Developers Blog
- Decoding cosmic signals with deep learning and Keras- Google Developers Blog
- Enterprise-Grade Precision for Long-Context Multimodal Embedding Inference on Cloud TPU- Google Developers Blog
- HeyGen x Google Cloud: Bringing Avatar IV to TPUs- Google Developers Blog