开源|Google发布Accelerator Agents：用Gemini将PyTorch工作负载迁至JAX并优化TPU内核

Google在GitHub开源了Accelerator Agents工具包，面向Google Cloud TPU开发场景，利用Gemini模型帮助开发者将PyTorch工作负载迁移到JAX，并优化自定义内核。该工具包包含MaxCode和MaxKernel两个核心模块：MaxCode负...

Read More