Skip to content
@NVIDIA-Merlin

NVIDIA-Merlin

Merlin is a framework providing end-to-end GPU-accelerated recommender systems, from feature engineering to deep learning training and deploying to production

Popular repositories

  1. Transformers4Rec Transformers4Rec Public

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

    Python 1k 142

  2. NVTabular NVTabular Public

    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 1k 143

  3. HugeCTR HugeCTR Public

    HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training

    C++ 893 193

  4. Merlin Merlin Public

    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 i…

    Python 687 110

  5. dataloader dataloader Public

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

    Python 396 24

  6. models models Public

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

    Python 241 48

Repositories

Showing 10 of 15 repositories
  • HierarchicalKV Public

    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 99 Apache-2.0 21 3 4 Updated Apr 28, 2024
  • Transformers4Rec Public

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

    Python 1,034 Apache-2.0 142 79 (3 issues need help) 5 Updated Apr 28, 2024
  • Merlin Public

    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 687 Apache-2.0 110 191 15 Updated Apr 28, 2024
  • models Public

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

    Python 241 Apache-2.0 48 137 11 Updated Apr 28, 2024
  • core Public

    Core Utilities for NVIDIA Merlin

    Python 18 Apache-2.0 14 18 8 Updated Apr 26, 2024
  • systems Public

    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 82 Apache-2.0 29 29 7 Updated Apr 26, 2024
  • NVTabular Public

    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,008 Apache-2.0 143 190 (1 issue needs help) 11 Updated Apr 23, 2024
  • HugeCTR Public

    HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training

    C++ 893 Apache-2.0 193 15 3 Updated Apr 21, 2024
  • dataloader Public

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

    Python 396 Apache-2.0 24 14 8 Updated Apr 16, 2024
  • distributed-embeddings Public

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

    Python 41 Apache-2.0 11 6 2 Updated Oct 17, 2023