@NVIDIA-Merlin
Guthub

NVIDIA-Merlin

Twitter logo Share LinkedIn logo Share Facebook logo Share
  • Stars
    star
    4,993
  • Global Org. Rank 4,603 (Top 2 %)
  • Registered over 3 years ago
  • Most used languages
    Python
         64.3 %
    C++
         14.3 %
    Jupyter Notebook
         14.3 %
    Cuda     7.1 %

Top repositories

1

Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
Python
1,076
star
2

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.
Python
1,030
star
3

HugeCTR

HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
C++
947
star
4

Merlin

NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.
Python
735
star
5

dataloader

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

models

Merlin Models is a collection of deep learning recommender system model reference implementations
Python
253
star
7

competitions

Solutions to Recommender Systems competitions
Jupyter Notebook
196
star
8

HierarchicalKV

HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of HierarchicalKV is to store key-value feature-embeddings on high-bandwidth memory (HBM) of GPUs and in host memory. It also can be used as a generic key-value storage.
Cuda
125
star
9

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.
Python
88
star
10

publications

Jupyter Notebook
61
star
11

distributed-embeddings

distributed-embeddings is a library for building large embedding based models in Tensorflow 2.
Python
42
star
12

gcp-ml-ops

MLOps pipeline for NVIDIA Merlin on GKE
Python
41
star
13

core

Core Utilities for NVIDIA Merlin
Python
19
star
14

nvtabular_triton_backend

Triton Backend for NVTabular
C++
2
star