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.
☆208Oct 28, 2026Updated this week
Alternatives and similar repositories for HierarchicalKV
Users that are interested in HierarchicalKV are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Examples for Recommenders - easy to train and deploy on accelerated infrastructure.☆359Updated this week
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,072Aug 3, 2026Updated 2 months ago
- Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.☆635Sep 4, 2025Updated last year
- A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster☆162Apr 20, 2024Updated 2 years ago
- ☆674Updated this week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A unified architecture deep learning framework designed specifically for ultra-large-scale sparse models.☆367Sep 23, 2026Updated last month
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,597Updated this week
- Fast hierarchical embedding cache for recommenders☆30Oct 20, 2026Updated last week
- DeepRec is a high-performance recommendation deep learning framework based on TensorFlow. It is hosted in incubation in LF AI & Data Foun…☆1,198Jan 21, 2025Updated last year
- GLake: optimizing GPU memory management and IO transmission.☆502Mar 24, 2025Updated last year
- Pytorch domain library for recommendation systems☆2,619Updated this week
- CUDA Embedding Lookup Kernel Library☆50Oct 9, 2026Updated 3 weeks ago
- ☆151Jan 9, 2025Updated last year
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆410Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A high-performance serving system for DeepRec based on TensorFlow Serving.☆20Nov 15, 2023Updated 2 years ago
- NCCL Profiling Kit☆156Jul 1, 2024Updated 2 years ago
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆932Dec 30, 2024Updated last year
- RDMA and SHARP plugins for nccl library☆232Aug 12, 2026Updated 2 months ago
- ☆24Oct 31, 2023Updated 3 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆46Oct 17, 2023Updated 3 years ago
- A std::execution style runtime context and High Performance RPC Transport for using OpenUCX. Including CUDA/ROCM/... devices with RDMA.☆34Oct 14, 2026Updated 2 weeks ago
- GPTQ inference TVM kernel☆41Apr 25, 2024Updated 2 years ago
- Herald: Accelerating Neural Recommendation Training with Embedding Scheduling (NSDI 2024)☆24May 9, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆47Updated this week
- A high performance and generic framework for distributed DNN training☆3,718Oct 3, 2023Updated 3 years ago
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆55Jul 3, 2022Updated 4 years ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆481Mar 15, 2024Updated 2 years ago
- Repository hosting code for "Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendations" (…☆2,017Updated this week
- End to End steps for adding custom ops in PyTorch.☆24Aug 20, 2020Updated 6 years ago
- ☆27Feb 17, 2025Updated last year
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- An easy-to-use framework for large scale recommendation algorithms.☆482Updated this week
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆27Feb 20, 2024Updated 2 years ago
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,376Aug 28, 2025Updated last year
- Microsoft Collective Communication Library☆400Aug 25, 2026Updated 2 months ago
- Code for reproducing results for SOSP paper Bagpipe☆12Oct 20, 2023Updated 3 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Jun 21, 2026Updated 4 months ago
- Universal Presentation: A Header-only C++ Library to Cout STL containers and more☆18Aug 14, 2023Updated 3 years ago
- a TensorFlow-based distributed training framework optimized for large-scale sparse data.☆335Apr 10, 2026Updated 6 months ago