NVIDIA-Merlin
Change the repository type filter
All
Repositories list
15 repositories
- 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.
- 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.
Transformers4Rec
PublicTransformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
- 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.
core
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.
models
PublicMerlin Models is a collection of deep learning recommender system model reference implementationsdataloader
PublicThe merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAXdistributed-embeddings
Publiccompetitions
Publicgcp-ml-ops
Public