HeavyBall Research is a research group spanning NYU Shanghai and NYU Tandon, led by Yucheng Lu.
Our mission is to build the next generation of AI systems. We work broadly on efficient modeling, hardware-aware training and inference algorithms, and the systems that bring them together at scale.
We are hiring!
Group Updates
- [2026-10] Thanks Amazon for supporting our research through the 2026 Amazon Research Award!
- [2026-09] We have released flywheel-tpu, a high-performance kernel library for TPUs. Read our story here!
- [2026-09] Multi-token Residual Prediction is accepted by NeurIPS’26, congrats to the team!
- [2026-04] Thanks Google for supporting our research through the 2026 Google TPU Research Award!
- [2026-04] We are delighted to welcome three new PhD students (Zishuo, Yufeng, and Runyu) to our group!