Towards Data Science By Chien Vu Minh

AI Agents Beat PyTorch: Writing Faster CUDA Kernels

Read the original on Towards Data Science →

AI agents have been shown to write CUDA kernels that outperform PyTorch, but validating these speedups is challenging. The author tested the agents on an NVIDIA DGX Spark and discovered that benchmark design is as crucial as the code itself. The article highlights the need for careful evaluation when comparing AI-generated performance claims.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at Towards Data Science.

arXiv AI
Oct 31

A Case Study of a Multi-Agent Kernel-Optimization Loop That Learns a New AI Accelerator

The article describes a case study where a large language model (LLM) agent loop was used to optimize three Gemma‑4‑12B decode kernels for the FuriosaAI RNGD NPU, a sparsely documented Rust DSL accelerator. Over 12 days on a shared device, the loop built its own target knowledge, performed microbenchmarks, recorded compiler legality rules, and iteratively refined kernel designs through role‑based search and acceptance tests. The resulting kernels achieved a 9.293× geometric mean speedup over the official baseline, each within 1.25× of the theoretical cycle minimum for weight movement.

By Byonggon Chun
arXiv AI
Jul 29

Kernel Forge: An Agent Harness for LLM-based Generation and Optimization of CUDA Kernels

arXiv:2607. 24762v1 Announce Type: new Abstract: Machine learning models are increasingly embedded in everyday software, and most of their runtime is spent in a small set of compute kernels such as matrix multiplication, convolution, and normalization.

By Joshua Brodsky, Dhravid Kumar, Savini Kashmira, Jayanaka Danatanarayana, Jason Mars, Krisztian Flautner, Lingjia Tang
arXiv Machine Learning
Jun 5

CUCo: An Agentic Framework for Compute and Communication Co-design

arXiv:2603. 02376v2 Announce Type: replace-cross Abstract: Computation and communication in distributed LLM training and inference are traditionally optimized in isolation; expert-crafted systems such as DeepEP, FLUX, and TokenWeave show the potential of co-design but require deep systems expertise and hardware-specific tuning; CUCo is an agentic framework that automates compute-communication co-design of CUDA kernels by combining a structured design-space formalization with a correctness-first fast-path agent for reliable baselines and an evolution-driven slow-path agent for high-performance strategies, achieving up to 1.

By Yoga Sri Varshan Varadharajan, Bodun Hu, Saurabh Agarwal, Aditya Akella