karlhigley

ranking-metrics-torch

Simple ranking metrics for PyTorch on CPU or GPU

torch-optim-sparse

PyTorch optimizers with sparse momentum and weight decay

practicalrecs-examples

Example code for practicalrecs.com blog posts

latus

Nearest neighbor search with integrated filtering

torch-factorization-models

Factorization-based models for recommender systems with PyTorch

py-pkg-template

A template for Python packages with poetry, pytest, Black, flake8, pylint, and pre-commit hooks

recommender_datasets

A common format and repository for various recommender datasets.

tensorflow

An Open Source Machine Learning Framework for Everyone

conda

OS-agnostic, system-level binary package manager and ecosystem

core

Core Utilities for NVIDIA Merlin

dataloader

The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX

DeepMatch

A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.

docs

Milvus web documents and contents

feast

Feature Store for Machine Learning

janusgraph-movielens

An ETL pipeline to augment the Movielens dataset and ingest it into Janusgraph

keras

Deep Learning for humans

milvus

An open source vector database powered by Faiss, NMSLIB and Annoy

models

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

mtg

Collection of data science and machine learning projects for Magic: the Gathering

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.

PySyft

A library for encrypted, privacy preserving deep learning

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

readme-md-rst-test

recodex

syft-proto

The JSON proto file which defines types for all Serde encoding across languages

systems

Merlin Systems provides tools for combining recommendation models with other elements of production recommender systems (like feature stores, nearest neighbor search, and exploration strategies) into end-to-end recommendation pipelines that can be served with Triton Inference Server.

Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation, available for both PyTorch and Tensorflow.

triton-inference-core

The core library and APIs implementing the Triton Inference Server.

triton-inference-server

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

witchhunt

A Python library for Approximate Nearest Necromancer (ANN) search