Alex L. Zhang
Incoming PhD Student at MIT CSAIL, Researcher @ VantAI, Princeton CS '24

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 ⏰):
- Neo-1, the first model for de novo molecular generation + structure prediction.
- Open source Project Popcorn 🍿 for generating GPU kernels with LLMs.
- KernelBench: environment for evaluating LLM-generated CUDA / Triton kernels.
- Multi-modal task instances for SWE-bench
with the original authors.
- Fast kernels for scaling Ligo's open source AlphaFold3, 900+ GitHub ⭐'s.
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:
- I first started programming by making Android games in Java.
- I used to make and sell PC games, one of which was mildly successful (~100k+ players). UE4 unfinished example.
- My favorite games of all time are Xenoblade Chronicles and Team Fortress 2.
latest posts
selected publications
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! |