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.
| Name | Name | Last commit date | ||
|---|---|---|---|---|
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.