Build and train a recommender system in 10 minutes using Keras and JAX- Google Developers Blog
May 13, 2025Yufeng Guo, Monica Song

Summary
Introduced Keras Recommenders (KerasRS), a new library for building state-of-the-art recommendation systems. KerasRS provides APIs with building blocks for ranking and retrieval tasks, supporting JAX, TensorFlow, and PyTorch backends. It includes specialized layers, losses, and metrics for recommender tasks, demonstrated with a `SequentialRetrievalModel` example using `BruteForceRetrieval` and `PairwiseHingeLoss`. Future plans include `DistributedEmbedding` for TPUs and continuous addition of model implementations.
More from the Predictive APIs and Machine Learning Integration trail
- Expanding Choice in Gemini Enterprise Agent Platform: Introducing Grounding with Parallel Web Search- Google Developers Blog
- Evolving Spec-Driven Development: Conductor Now Supports Antigravity- Google Developers Blog
- Building scalable AI agents with modular prompt transpilation- Google Developers Blog
- LiteRT.js, Google's high performance Web AI Inference- Google Developers Blog
- Gemma 4 12B: The Developer Guide- Google Developers Blog