PallasForge 开源:Pallas/JAX 写 GPU 内核的实战教程库
A_K_Nain · x · 2026-10-13
Google 工程师 Aakash Kumar Nain 开源了 PallasForge,一个用 JAX 的 Pallas 扩展编写与基准测试 GPU 内核的教程项目,旨在填补官方文档只覆盖少量通用示例的空白。
- 目标覆盖三类真实场景:面向 Hopper/Blackwell 等不同 GPU 家族的实战例子,如融合反量化、flash-attention 变体、fused-norm 等,而非仅教 matmul
- 教程配套性能剖析内容:演示用 Nsight Compute 深入内核内部,解读 GPU Speed of Light、内存负载、寄存器溢出、occupancy、warp stalls 等指标,并附示例报告讲解如何读数
- 仓库以 Apache-2.0 协议开源,作者欢迎社区贡献内核、基准测试与实现思路说明
「Infra」频道最新
- 大厂规模曝光:每天2500万次代码检出仅用于CI — mcraddock · 2026-10-14
- Cloudflare Agents SDK 推出持久化浏览器工具,会话跨轮次保留 — irvinebroque · 2026-10-14
- Ed Zitron 长文：AI 泡沫建立在拍脑袋的「年化营收」数字上 — GaryMarcus · 2026-10-14
- 曝博通early谈判为 OpenAI 筹逾500亿美元融资买定制芯片 — Beth_Kindig · 2026-10-14
- Ben Bajarin：埃米时代预测算力未来，看 WFE 设备路线图比芯片路线图更准 — BenBajarin · 2026-10-14
- AWS 原生搬来 Vercel Fluid Compute,Lambda 推出 Web Functions — cramforce · 2026-10-14