Alex L. Zhang

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 the Princeton CS department’s highest academic honor awarded to a single student.

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:

• Open source Project Popcorn 🍿 for generating GPU kernels with LLMs.
• Multi-modal task instances for SWE-bench with the original authors.
• Fast kernels for scaling Ligo's open source AlphaFold3.
• World models that can be conditioned on language for model-based agents.
• Simple FlashAttention2 with custom masks in Triton.

Feel free to reach out to talk through my email at [x]@princeton.edu where [x]=alzhang.

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 PC games, one of which (embarrassingly) was mildly popular (~100k+ players). UE4 unfinished example.
• My favorite game of all time is Xenoblade Chronicles.