A small library for recommendation systems written in JAX/Flax/Linen.

See recommenders/cfdnn/train_movielens_cfdnn.py for a simple example of a Collaborative Filtering Deep Neural Network recommender applied to MovieLens.