HierarchicalKV
HierarchicalKV是NVIDIA Merlin项目的组成部分,为推荐系统提供分层键值存储功能。该库可在GPU高带宽内存和主机内存中存储特征嵌入,支持大规模推荐模型训练。通过绕过CPU和实现表大小约束策略,HierarchicalKV提升了性能和内存利用率。这使得NVIDIA GPU更适合训练大型搜索、推荐和广告模型,简化了复杂推荐模型的构建、评估和部署过程。