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

Hey ð! 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!
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 research (with ð's!) I've led / majorly contributed to (ð by â°):
Benchmarks ðª:
- VideoGameBench: environment for evaluating VLMs on popular video games!
- KernelBench: environment for evaluating LLM-generated CUDA / Triton kernels.
- SWE-bench Multimodal
: Multi-modal task instances for SWE-bench with the original team.
Models ð:
- Neo-1: the first deep learning model for de novo molecular generation + structure prediction.
- Language-guided World Models: world model RL agents that can be conditioned on natural language hints.
MLSys â¡:
- AMD x GPU MODE $100k MI300X GPU kernel competition: write the fastest kernels on MI300s!
- Project Popcorn ð¿: open source effort to build LLM-generated GPU kernels.
- Fast Triton kernels for Ligo Biosciences' open-source AlphaFold3 (950+ â'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!
Random. 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! |