Google

Using KerasHub for easy end-to-end machine learning workflows with Hugging Face- Google Developers Blog

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

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