CV
Contact Information
| Name | Weian Mao |
| Professional Title | Postdoctoral Researcher |
| weian@mit.edu |
Professional Summary
Postdoctoral researcher at MIT CSAIL working on LLMs, VLMs, and AI for Science.
Experience
-
2025 - present Cambridge, MA, US
Postdoctoral Associate
Massachusetts Institute of Technology
Working with Prof. Regina Barzilay at MIT CSAIL.
- Research on LLMs, VLMs, and AI for Science
- KV cache compression for long-context LLM reasoning
Education
-
2020 - 2024 Adelaide, Australia
Ph.D.
University of Adelaide
Computer Science
- Advised by Prof. Chunhua Shen
- Australian Institute for Machine Learning (AIML)
-
2015 - 2019 Shanghai, China
B.Eng.
East China University of Science and Technology
Electrical Engineering
Skills
Frameworks (): PyTorch, NVIDIA NeMo, Megatron-LM, vLLM, verl, DeepSpeed, FSDP, Hugging Face Transformers
Technical (): Distributed Training (multi-node, multi-GPU), RLHF/RLVR, Post-training Optimization, KV Cache Compression, Model Alignment
Research (): Large Language Models, Vision-Language Models, Reinforcement Learning for LLMs, Efficient Inference, Protein Design