NVIDIA-Merlin
Change the repository type filter
All
Repositories list
15 repositories
HierarchicalKV
PublicHierarchicalKV 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.Merlin
PublicNVIDIA 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 is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
HugeCTR
PublicHugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating trainingpublications
PublicNVTabular
PublicNVTabular 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..github
Publiccore
Publicsystems
PublicMerlin 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
Publicgcp-ml-ops
Public