Alex L. Zhang

Incoming PhD Student at MIT CSAIL, Researcher @ VantAI, Princeton CS '24

Alex_Zhang_2024.png

The first thing you should do is check out vgbench.com, and then go participate in the AMD x GPU MODE $100K GPU coding competition!

Hey 👋! I recently graduated after spending a wonderful 4 years at Princeton University advised by Professor Karthik Narasimhan, Dr. Khanh Nguyen, and Professor Kai Li. There, I received the Phillip Goldman ‘86 Senior Prize in Computer Science, which is awarded to the top undergraduate student in the Princeton CS department.

The most recent projects (+ research) I've led / majorly contributed to are (🔀 by ⏰):

Feel free to reach out to talk through my email at [x]@princeton.edu where [x]=alzhang. I’m also very active in the GPU MODE community and co-lead the GPU programming leaderboard!


Right now, I’m particularly interested in systems for/in ML, scaling and understanding language models, and applications of AI to the natural sciences!

I got into AI research because I wanted to design intelligent bots for video games when I was younger:


latest posts


selected publications

  1. NLP
    kernelbench.png
    KernelBench: Can LLMs Write Efficient GPU Kernels?
    Anne Ouyang*, Simon Guo*, Simran Arora, Alex Zhang , and 3 more authors
    DL4C (Best Paper Award), SSI-FM Workshop @ ICLR 2025
  2. 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
  3. NLP
    lgwm.png
    Language-guided World Models: A Model-Based Approach to AI Control.
    Alex Zhang*, Khanh Nguyen*, Jens Tuyls, Albert Lin , and 1 more author
    SpLU-RoboNLP Workshop @ ACL 2024 (Oral)
  4. Dataset
    asap.png
    Building Scalable Video Understanding Benchmarks through Sports
    Aniket Agarwal*Alex Zhang*, Karthik Narasimhan, Igor Gilitschenski , and 2 more authors
    DMLR Workshop @ ICLR 2024

news

Nov 01, 2024 Started as a researcher at VantAI under the wonderful Luca Naef!
Sep 17, 2024 Recently built a fast and memory-efficient Triton kernel to fuse the MSA Weighted Pair Averaging with Gating algorithm in AlphaFold3 for an awesome open-source implementation.
Jun 03, 2024 Starting a research internship at Snapchat focused on large-scale recommendation systems!
Jun 01, 2023 Started working @ stealth startup on some zany generative AI stuff!
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!