热门
导航
快讯
推荐文章
热门
导航
快讯
推荐文章
#TorchRec
torchrec - 旨在提供大规模推荐系统所需的常见稀疏性和并行性原语的PyTorch库
TorchRec
PyTorch
推荐系统
CUDA
FBGEMM
Github
开源项目
TorchRec是一个专为大规模推荐系统设计的PyTorch库,提供稀疏性和并行性解决方案。它支持多种嵌入表分片策略,并能自动优化分片计划。通过流水线训练和优化内核,提高模型性能。还支持量化训练和推理,包含多个验证的模型架构和数据集示例,适用于需要高性能和扩展性的推荐系统项目。
1
1
相关文章
TorchRec: 构建大规模推荐系统的PyTorch领域库
3 个月前
使用协议
隐私政策
广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI
·
鲁ICP备2024100362号-6
·
鲁公网安备37021002001498号