HeyGen, an AI video generation platform, announced that its Avatar IV diffusion model, which runs on more than 18B parameters, is now 1.86× faster on Google Cloud's eight-chip Trillium (v6e) TPU host …
JAX's Pallas kernel system, which lowers through Triton on GPU and Mosaic on TPU, lets developers write custom kernels when XLA's automatic fusion is insufficient for operations like flash attention, …
Google's open-source TPU compiler, shipped as the Mosaic TPU dialect inside JAX, documents eight explicit memory spaces that programmers must manage, including vector memory (VMEM), scalar memory (SME…
Helion, PyTorch's high-level DSL for writing performance-portable ML kernels, partnered with Google to build a TPU backend that compiles Helion kernels to Pallas, enabling PyTorch-friendly TPU kernel …
Google launched the TPU Developer Hub, a centralized educational resource for guidance on Google Cloud TPU infrastructure and software stack. The hub covers the full developer lifecycle from pre-train…
Google Cloud launched the TPU Developer Hub, a centralized educational resource providing technical guidance on TPU hardware, software stack, and optimization strategies for AI developers. The hub off…
NVIDIA's C++ template-based CUTLASS library for GPU kernels suffers from compile times of up to 20 seconds for a single kernel and over 17 minutes for full builds, prompting a shift toward Python-embe…