PyTorch Blog· Meta Team: Daohang Shi, Oleksandr Stashuk, Rupert Wu, Liangbei Xu, Rich Zhu·· 5 小时前AI 评分41
PyTorch 用 FBTriton 内核重构 Table Batched Embeddings
Modernizing Table Batched Embeddings with FBTriton
AI 导读
PyTorch 推出 FBTriton 内核，用 Triton 实现 Table Batched Embedding（TBE）的前向与反向传播，替代推荐系统中跨数千块分片 GPU 的旧版 CUDA 内核。
来源：PyTorch Blog · pytorch.org