Google open-sources TPU inference tool Raiden, competing with NVIDIA NIXL
Google open-sourced Raiden, a TPU inference tool that directly competes with NVIDIA's NIXL, featuring efficient KVCache transfer between prefill and decode instances, KVCache offloading, and compatibility with JAX and PyTorch.
Detected & updated continuously · Source: Nebula
Sources
@NFT_Chen
🔥 重磅炸裂!Google 开源 TPU 推理神器 Raiden! 直接对标 NVIDIA NIXL,TPU 软件栈再放大招! 关键亮点速览: • Prefill ↔ Decode 实例间 KVCache 超高效传输 • 完整支持 KVCache 卸载 + 移动原语 • 主机侧共享内存持久化,重启零冷启动 • 同时兼容 JAX & PyTorch • Google https://t.co/Y5kJt0r3eZ
@linasbeliunas
Google’s former Chief Scientist Jeff Dean explaining how the idea behind TPUs was born at Google: https://t.co/p377Vopnhp