Alex L. Zhang

PhD Student at MIT CSAIL, Princeton CS '24

az.png

Hi! I broadly work on areas where language models are underutilized or inefficient. I am currently a PhD student at MIT CSAIL advised by Omar Khattab and Tim Kraska.

Some notable things worth mentioning:

I’ve had brief but pleasant stints at a variety of companies: Sakana AI, VantAI, Snap Research, Apple, and Claryo.


research highlights:


Feel free to reach out to talk about anything through my email at [x]@mit.edu where [x]=altzhang. Finally, I’m also fairly active in and help moderate the GPU MODE community. As a last fun fact, I used to make and sell PC games, one of which was mildly successful (~100k+ players). [unfinished example]


latest posts


selected publications

  1. rlm.png
    Recursive Language Models
    Alex Zhang, Tim Kraska, and Omar Khattab
    arXiv
  2. kernelbot.png
    KernelBot: A Competition Platform for Writing Heterogeneous GPU Code
    Alex Zhang*, Matej Sirovatka*, Erik Schultheis*, Benjamin Horowitz* , and 1 more author
    CODEML @ ICML 2025 (Spotlight)
  3. videogamebench.png
    VideoGameBench: Can Vision-Language Models complete popular video games?
    Alex Zhang, Thomas L. Griffiths, Karthik R. Narasimhan, and Ofir Press
    Under review
  4. NLP
    kernelbench.png
    KernelBench: Can LLMs Write Efficient GPU Kernels?
    Anne Ouyang*, Simon Guo*, Simran Arora, Alex Zhang , and 3 more authors
    ICML 2025, DL4C (Best Paper) & SSI-FM Workshop @ ICLR 2025
  5. NLP
    teaser_mm.png
    SWE-bench Multimodal: Do AI Systems Generalize to Visual Software Domains?
    John Yang*, Carlos E. Jimenez*Alex Zhang, Kilian Lieret , and 9 more authors
    ICLR 2025

news

Jun 12, 2025 Went on stage with Lisa Su, CEO of AMD, who recognized our AMD GPU programming competition and announced the winners! [picture].
Nov 01, 2024 Started as a researcher at VantAI under the wonderful Luca Naef!
Jun 03, 2024 Starting a research internship at Snapchat focused on large-scale recommendation systems!
Mar 17, 2023 Organized and went on AI TigerTrek with my close (and brilliant!) friends Evan Dogariu, Michael Tang, and Jiatong Yu! We visited and had Q&A’s at OpenAI, Google, Anthropic, Redwood Research, Stanford HAI, Nuro, and more!
Aug 29, 2022 Finished my internship in Seattle at Apple INFI working on LMs in Siri!