arXiv Machine Learning

From GPU to TPU with Coding Agents: What Works, What Fails, and What It Costs?

The study investigates whether coding agents can reliably migrate PyTorch machine learning projects from GPUs to TPUs. Using Claude Code, ten projects were migrated, with eight meeting the project‑goal criterion but 76% of execution attempts failing, often due to compilation issues or silent changes in computation. The research highlights significant human effort (average 2.4 interventions) and variable costs, noting that while some TPUs can be faster and cheaper, failures and accuracy drops remain common.

arXiv Machine Learning
Aug 14

CAKE: Compiler-Agent Co-Design for Frontier Kernel Evolution

arXiv:2608. 12629v1 Announce Type: new Abstract: GPU kernel agents and GPU programming languages have advanced separately, leaving expert kernels difficult to reproduce.

By Zihao Ye, Yingyi Huang, Hongyi Jin, Bohan Hou, Junru Shao, Zhongming Yu, Jinqi Chen, Meghan Cowan, Shiyi Cao, Shanli Xing, Hanfeng Chen, Vinod Grover, Tianqi Chen, Luis Ceze
arXiv AI
Sep 16

Ave: Guiding Agentic GPU Optimization Using Data-Flow Invariants

arXiv:2604.18616v2 Announce Type: replace-cross Abstract: LLM coding agents can generate correct GPU kernels, but their performance still trails expert libraries. Reaching peak throughput requires co...

By Haohui Mai, Xiaoyan Guo, Xiangyun Ding, Daifeng Li, Qiuchu Yu, Chenzhun Guo, Cong Wang, Jiacheng Zhao, Christos Kozyrakis, Binhang Yuan
arXiv Computation and Language
Oct 22

TrainBridge: Agentic NPU Training Optimization Guided by GPU Compilation Evidence

TrainBridge is an agentic approach that transfers optimization guidance from GPU training to Ascend NPU training. It extracts optimization intents from highly optimized GPU execution and uses them to rewrite NPU programs, then refines these rewrites through iterative profiling. Experiments show that TrainBridge yields an average 1.7× speedup across four workloads and nearly 2.0× speedup on 27 Modded‑NanoGPT speedrun records.

By Yupeng Chen, Kunxiang Zhao, Xingjian Wang, Yue Zhao, Yijing Wu, Ye Liu, Kai Li, Congliang Chen, Ruoyu Sun
arXiv AI
Oct 16

Agent Skill Framework: Perspectives on the Potential of Small to Medium Language Models in Industrial Environments

The paper investigates how the skill paradigm—using modular abilities within agentic frameworks—works with small to medium open‑source language models (270 M–80 B parameters) in industrial, resource‑constrained settings. A large‑scale empirical study on an insurance‑claim classification task and a set of Python programming problems shows that very small models often fail to retrieve or select correct skills, while larger 30 B–80 B models benefit significantly from skill augmentation. The authors also present a deployment‑oriented GPU profile, reporting VRAM time occupancy (GB·min) across models, to help characterize GPU‑memory costs for different model scales. "whyItMatters":"The study provides concrete evidence on how skill augmentation scales with model size, task complexity, and GPU budget, offering practical deployment insights for industrial applications using open‑source language models."

By Yangjie Xu, Lujun Li, Lama Sleem, Niccolo Gentile, Yewei Song, Yiqun Wang, Siming Ji, Wenbo Wu, Hui Huang, Radu State
arXiv AI
Sep 11

KernelGenBench: Can LLMs and Agents Write Efficient Kernels Across Operator Sources and Hardware Platforms?

KernelGenBench is a unified benchmark that evaluates large language models and agentic systems for generating efficient Triton kernels across diverse operator sources and hardware platforms. It covers 210 operators from PyTorch ATen, vLLM, and cuBLAS, and tests a 110‑operator subset on six different chips, consuming over 15 billion tokens in evaluation. The study finds that no single method dominates across all sources and platforms, with significant variations in correctness and performance depending on the operator source and hardware, and that agentic approaches require millions of tokens per successful operator.

By Peiyu Zang, Jian Tao, Jialing Zhang, Yichen Yuan, Wentao Zhang, Guang Liu, Yonghua Lin
arXiv AI
Oct 28

InfraDev: Can Coding Agents Develop AI Infrastructure, and How Can We Tell?

InfraDev is a new benchmark comprising 36 tasks derived from real-world pull requests to Megatron‑LM, vLLM, and verl, designed to test coding agents on infrastructure development for training, inference, and reinforcement learning across many GPUs. Each task requires agents to reimplement multi‑file changes and run correctly, with grading that includes hidden probes and a prompt exposing only the interfaces used. The benchmark reveals that current frontier models resolve only about 39% of tasks, with probes rejecting a third of submissions that pass official tests, and that structural and state errors dominate failures.

By Yangshen Deng, Jiaqi Ruan, Yeqi Huang, Nick Lau, Guangming Sheng, Luo Mai