Predictive APIs and Machine Learning Integration
Build and train a recommender system in 10 minutes using Keras and JAX- Google Developers Blog

Build and train a recommender system in 10 minutes using Keras and JAX- Google Developers Blog

5/13/2025 · Yufeng Guo, Monica Song

What this post added

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.

Read the original post ↗