#NVIDIA Merlin
HierarchicalKV: NVIDIA的高性能分层键值存储解决方案
3 个月前
相关项目
Merlin
Merlin是NVIDIA开发的开源库，为推荐系统提供GPU加速解决方案。它包含多个组件如NVTabular和HugeCTR，支持大规模数据处理、特征工程、模型训练和部署。Merlin能处理数百TB数据，通过GPU加速提升系统性能。它兼容TensorFlow、PyTorch等框架，便于构建和优化推荐模型。
HierarchicalKV
HierarchicalKV是NVIDIA Merlin项目的组成部分，为推荐系统提供分层键值存储功能。该库可在GPU高带宽内存和主机内存中存储特征嵌入，支持大规模推荐模型训练。通过绕过CPU和实现表大小约束策略，HierarchicalKV提升了性能和内存利用率。这使得NVIDIA GPU更适合训练大型搜索、推荐和广告模型，简化了复杂推荐模型的构建、评估和部署过程。