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.
☆194Feb 27, 2026Updated this week
Alternatives and similar repositories for HierarchicalKV
Users that are interested in HierarchicalKV are comparing it to the libraries listed below
Sorting:
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,048Sep 15, 2025Updated 5 months ago
- Examples for Recommenders - easy to train and deploy on accelerated infrastructure.☆219Feb 13, 2026Updated 2 weeks ago
- A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster☆161Apr 20, 2024Updated last year
- An Attention Superoptimizer☆22Jan 20, 2025Updated last year
- ☆625Feb 20, 2026Updated last week
- ☆152Jan 9, 2025Updated last year
- Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.☆632Sep 4, 2025Updated 5 months ago
- NCCL Profiling Kit☆152Jul 1, 2024Updated last year
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆918Dec 30, 2024Updated last year
- Benchmark tests supporting the TiledCUDA library.☆18Nov 19, 2024Updated last year
- DeepRec is a high-performance recommendation deep learning framework based on TensorFlow. It is hosted in incubation in LF AI & Data Foun…☆1,168Jan 21, 2025Updated last year
- End to End steps for adding custom ops in PyTorch.☆24Aug 20, 2020Updated 5 years ago
- GLake: optimizing GPU memory management and IO transmission.☆498Mar 24, 2025Updated 11 months ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆47Oct 17, 2023Updated 2 years ago
- TiledLower is a Dataflow Analysis and Codegen Framework written in Rust.☆14Nov 23, 2024Updated last year
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,535Updated this week
- DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit☆92Jan 26, 2026Updated last month
- An experimental parallel training platform☆56Mar 25, 2024Updated last year
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆478Mar 15, 2024Updated last year
- RDMA and SHARP plugins for nccl library☆224Jan 12, 2026Updated last month
- ☆25Feb 20, 2024Updated 2 years ago
- Pytorch domain library for recommendation systems☆2,477Updated this week
- Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt, but ported to both NVIDIA and AMD GPUs.☆32Apr 2, 2025Updated 11 months ago
- Universal Presentation: A Header-only C++ Library to Cout STL containers and more☆18Aug 14, 2023Updated 2 years ago
- Microsoft Collective Communication Library☆385Sep 20, 2023Updated 2 years ago
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆411Jun 14, 2025Updated 8 months ago
- ☆47Dec 13, 2024Updated last year
- sgx-based encrypted deduplication prototype☆14May 14, 2021Updated 4 years ago
- ☆20Sep 28, 2024Updated last year
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆475Updated this week
- OpenAI compatible API for open source LLMs☆16Oct 30, 2023Updated 2 years ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆44Feb 27, 2025Updated last year
- A high performance and generic framework for distributed DNN training☆3,716Oct 3, 2023Updated 2 years ago
- Mooncake is the serving platform for Kimi, a leading LLM service provided by Moonshot AI.☆4,843Updated this week
- Nsight Compute In Docker☆13Dec 21, 2023Updated 2 years ago
- A high-performance distributed deep learning system targeting large-scale and automated distributed training.☆333Dec 13, 2025Updated 2 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Feb 20, 2026Updated last week
- Passive Disaggregated Persistent Memory at USENIX ATC 2020.☆51Nov 2, 2020Updated 5 years ago
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆56Jul 3, 2022Updated 3 years ago