Google quietly released a powerful recommender systems library optimized for JAX and TPUs, based on Keras. It's called RecML. It has native support for SparseCore (latest hardware for handling large distributed embeddings)

→ View original post on X — @fchollet