Product
Pallas
A low-level programming interface within the JAX ecosystem for writing custom TPU and GPU kernels. It allows developers to control memory layout and computation scheduling to achieve maximum performance optimization.
Timeline (last 90 days)
Sep 24
All 92 MoE layers of Kimi K3 run within a single Pallas kernel.
Unverified50%
Sep 24
Pallas is a programming interface in the JAX ecosystem for writing custom TPU/GPU kernels.
Unverified50%