Google

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

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

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