产品
Pallas
JAX生态中用于编写自定义TPU/GPU内核的低级编程接口,允许开发者控制内存布局与计算调度以实现极致性能优化
时间轴 (近 90 天)
9月24日
Kimi K3 的全部 92 个 MoE 层都运行在单一的 Pallas 内核中
待验证50%
9月24日
Pallas 是 JAX 生态里用于编写自定义 TPU/GPU 内核的编程接口
待验证50%
JAX生态中用于编写自定义TPU/GPU内核的低级编程接口,允许开发者控制内存布局与计算调度以实现极致性能优化
Kimi K3 的全部 92 个 MoE 层都运行在单一的 Pallas 内核中
Pallas 是 JAX 生态里用于编写自定义 TPU/GPU 内核的编程接口