Alex L. Zhang
Researcher @ VantAI, Princeton CS '24

Hey 👋! I graduated from 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.
Right now, I’m particularly interested in systems for/in ML, scaling and understanding language models, and applications of AI to the natural sciences!
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!
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
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! |
Aug 29, 2022 | Finished my internship in Seattle at Apple INFI working on LMs in Siri! |