PallasForge 开源：用 Pallas/JAX 写 GPU 内核的实战教程库

Google 工程师 Aakash Kumar Nain 开源了 PallasForge，一个用 JAX 的 Pallas 扩展编写与基准测试 GPU 内核的教程项目，意在填补官方文档只覆盖少量通用示例的空白。