NVIDIA-Merlin
Popular repositories Loading
-
Transformers4Rec
Transformers4Rec PublicTransformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
-
dataloader
dataloader PublicThe merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
Repositories
- publications Public
- 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.
- Transformers4Rec Public
Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
- 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.