Recommender System Python Packages

lancedb

Developer-friendly OSS embedded retrieval library for multimodal AI. Search More; Manage Less.

torchrec

Pytorch domain library for recommendation systems

usearch

Fast Open-Source Search & Clustering engine × for Vectors & Arbitrary Objects × in C++, C, Python, JavaScript, Rust, Java, Objective-C, Swift, C#, GoLang, and Wolfram 🔍

implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets

tensorflow-recommenders

TensorFlow Recommenders is a library for building recommender system models using TensorFlow.

lightfm

A Python implementation of LightFM, a hybrid recommendation algorithm.

torchrec-nightly

Pytorch domain library for recommendation systems

catalyst

Accelerated deep learning R&D

rectools

RecTools - library to build Recommendation Systems easier and faster than ever before

rusket

rusket 🦀🧺

usearch-iscc

Fast Open-Source Search & Clustering engine × for Vectors & Arbitrary Objects × in C++, C, Python, JavaScript, Rust, Java, Objective-C, Swift, C#, GoLang, and Wolfram 🔍

librecommender

Versatile End-to-End Recommender System

nvtabular

NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale datasets used to train deep learning based recommender systems.

merlin-models

Merlin Models is a collection of deep learning recommender system model reference implementations

torchrec-nightly-cpu

Pytorch domain library for recommendation systems

replay-rec

A Comprehensive Framework for Building End-to-End Recommendation Systems with State-of-the-Art Models

transformers4rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.

mmsbm

Compute Mixed Membership Stochastic Block Models.

persia

High performance distributed framework for training deep learning recommendation models based on PyTorch.

buffalo

TOROS Buffalo: A fast and scalable production-ready open source project for recommender systems

spotify-recommender-api

Introducing "spotify-recommender-api", a pip package that uses an adaptation of the KNN algorithm to make personalized song recommendations for users. It analyzes your Spotify listening stats and suggests new tracks you may enjoy, and also clusters similar songs on a big playlist in order to help one find songs that play well in a smaller playlist

rexmex

A general purpose recommender metrics library for fair evaluation.

matrix-factorization

Library for matrix factorization for recommender systems using collaborative filtering