PyTorch

Bringing PyTorch Monarch to AMD GPUs: Single-Controller Distributed Training on ROCm

Training state-of-the-art large language models (LLMs) with billions of parameters requires distributed training across hundreds or thousands of GPUs. At this scale, hardware failures are not exceptional events—they are expected....

PyTorch 2.12 Release Blog

We are excited to announce the release of PyTorch® 2.12 (release notes)!   The PyTorch 2.12 release features the following changes:  Batched linalg.eigh on CUDA is up to 100x faster due...

In-Kernel Broadcast Optimization: Co-Designing Kernels for RecSys Inference

TL;DR: Traditional RecSys inference explicitly replicates shared user embeddings/sequences for every candidate. In-Kernel Broadcast Optimization (IKBO) eliminates this overhead via a kernel-model-system co-design that fuses broadcast logic directly into user-candidate...