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

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

6/24/2025 · Yufeng Guo, Divyashree Sreepathihalli, Monica Song

What this post added

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.

Read the original post ↗