PAIR-code/jax-recommenders

JAX Recommender Systems Library

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.