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 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:

• 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, 900+ GitHub ⭐'s.
• World models that can be conditioned on language for model-based RL 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 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.