← Search

Alex L Zhang

2 accepted papers

2025

KernelBench: Can LLMs Write Efficient GPU Kernels?

ICML 2025poster

Efficient GPU kernels are crucial for building performant machine learning architectures, but writing them is a time-consuming challenge that requires significant expertise; therefore, we explore using language models (LMs) to automate kernel generation. We introduce **KernelBench**, an open-source…

Cited by 1SourcePDFScholar
2025

SWE-bench Multimodal: Do AI Systems Generalize to Visual Software Domains?

ICLR 2025poster

Autonomous systems for software engineering are now capable of fixing bugs and developing features. These systems are commonly evaluated on SWE-bench (Jimenez et al., 2024a), which assesses their ability to solve software issues from GitHub repositories. However, SWE-bench uses only Python repositor…

Cited by 198SourcePDFScholar