About
Hi! I am a first-year PhD student in Computer Science at National University of Singapore advised by Yang You, where I also completed my master’s studies. I obtained my bachelor’s degree in CS & EE from Huazhong University of Science and Technology supervised by Xinggang Wang. Previously, I collaborated at Pika with Chenlin Meng and interned at Colossal-AI with Jiarui Fang.
I am actively looking for summer research internship in 2025. Please feel free to reach out if there are any opportunities available.
Research
My current research mainly focuses on efficient and scalable machine learning systems through parallelism, algorithm, scheduling and compiler and optimization, recently with a primary emphasis on video models acceleration. I am always happy to chat about interesting research ideas, and looking for academic collaborations. Please drop me an email if you are interested in collaborating with me.
Open-Source Projects
- VideoSys: An Easy and Efficient System for Video Generation
Project lead. - Colossal-AI: Making large AI models cheaper, faster and more accessible
Core contributor (ranked 5th by 2024) with 38k star.
Selected Publications (all)
System for Video Models
Real-Time Video Generation with Pyramid Attention Broadcast
| Arxiv | First author | paper | code | blog |DSP: Dynamic Sequence Parallelism for Multi-Dimensional Transformers
| Arxiv | First author | paper | code |
System with Low Memory Cost
AutoChunk: Automated Activation Chunk for Memory-Efficient Long Sequence Inference
| ICLR 2024 | First author | paper | code |HeteGen: Heterogeneous Parallel Inference for Large Language Models on Resource-Constrained Devices
| MLSys 2024 | First author | paper |
System for Machine Learning & Science
- FastFold: Optimizing AlphaFold Training and Inference on GPU Clusters
| PPoPP 2024 | Second author | paper | code |
Experience
Pika
Host: Chenlin Meng
Efficient video system.Colossal-AI
Research Engineer Intern
Host: Jiarui Fang and Shenggui Li
Core contributor (ranked 5th by 2024) with 38k star, and responsible for features including ZeRO, TP, PP, compiler, and MoE.
2022.07 - 2023.12
Education
- National University of Singapore
Ph.D. in Computer Science
2024.01 - now
M.S. in Computer Science
2022.08 - 2023.12 - Huazhong University of Science and Technology
B.S. in Computer Science & Electronic Information
2018.09 - 2022.06