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.

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

lightfm

A Python implementation of LightFM, a hybrid recommendation algorithm.

rusket

rusket 🦀🧺

catalyst

Accelerated deep learning R&D

torchrec-nightly

Pytorch domain library for recommendation systems

rectools

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

librecommender

Versatile End-to-End Recommender System

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 🔍

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.

persia

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

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

persia-cuda111

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

transformers4rec

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

persia-cuda113

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

persia-cuda102

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

mmsbm

Compute Mixed Membership Stochastic Block Models.