About

Hey, I am Ziming! I am a second-year PhD student at UC Berkeley, working with Prof. Ion Stoica and Prof. Scott Shenker. I am affiliated with Sky Computing and NetSys Lab. I completed my undergrad from Yale University, with double major in Computer Science and Philosophy.

Earlier, I was advised by Prof. Anurag Khandelwal, Prof. Dragomir Radev at Yale, and Prof. Rashmi Vinayak at CMU. I am fortunate to work with collaborators from Microsoft Research, NetApp, Facebook AI, Salesforce Research, Twitter, and Databricks. I was awarded Berkeley College of Engineering Fellowship (2023), Outstanding Undergraduate Researcher Award (2023) Runner-Up, Computer Science Prize (2023), and Rosenfeld Science Scholars Fellowship (2021). My works have received best-paper awards at EuroSys’24 and NSDI’24.

My research areas are broadly in distributed systems, cloud, data and AI systems. I am building SkyPilot, a framework for running LLMs, AI, and batch jobs on any cloud. I am also working with Atul Adya on distributed caches and consistency at Databricks.

Education

  • Ph.D. in Computer Science, UC Berkeley, 2023 - Present
  • B.S. in Computer Science and Philosophy, Yale University, 2019 - 2023

Publications

Revisiting Cache Freshness for Emerging Real-Time Applications
Ziming Mao, Rishabh Iyer, Scott Shenker, Ion Stoica
HotNets 2024
[Paper]

SkyServe: Serving AI Models across Regions and Clouds with Spot Instances
Ziming Mao*, Tian Xia*, Zhanghao Wu, Wei-Lin Chiang, Tyler Griggs, Romil Bhardwaj, Zongheng Yang, Scott Shenker, Ion Stoica
In Submission
[Paper] [Code]

Trinity: A Fast and Space-efficient Multi-attribute Data Store
Ziming Mao, Anurag Khandelwal, Kiran Srinivasan
EuroSys 2024, Best Student Paper Award
[Paper] [Code]

Can’t Be Late: Optimizing Spot Instance Savings under Deadlines
Zhanghao Wu, Wei-Lin Chiang, Ziming Mao, Zongheng Yang, Eric Friedman, Scott Shenker, Ion Stoica
NSDI 2024, Outstanding Paper Award
[Paper] [Code]

GL-Cache: Group-level Learning for Efficient and High-Performance Caching
Juncheng Yang, Ziming Mao, Rashmi Vinayak, Yao Yue
FAST 2023
[Paper] [Code]

Pie: Pooling CPU Memory for LLM Inference.
Yi Xu, Ziming Mao, Xiangxi Mo, Shu Liu, Ion Stoica
In Submission
[Paper]

DYLE: Dynamic Latent Extraction for Abstractive Long-Input Summarization
Ziming Mao*, Chen Henry Wu*, Ansong Ni, Yusen Zhang, Rui Zhang, Tao Yu, Budhaditya Deb, Chenguang Zhu, Ahmed Hassan Awadallah, Dragomir R. Radev
ACL 2022
[Paper] [Code]

SummN: A Multi-Stage Summarization Framework for Long Input Dialogues and Documents
Yusen Zhang, Ansong Ni, Ziming Mao, Chen Henry Wu, Chenguang Zhu, Budhaditya Deb, Ahmed Hassan Awadallah, Dragomir R. Radev, Rui Zhang
ACL 2022
[Paper] [Code]

FeTaQA: Free-form Table Question Answering
Linyong Nan, Chiachun Hsieh, Ziming Mao, Xi Victoria Lin, Neha Verma, Rui Zhang, Wojciech Kryściński, Nick Schoelkopf, Riley Kong, Xiangru Tang, Murori Mutuma, Ben Rosand, Isabel Trindade, Renusree Bandaru, Jacob Cunningham, Caiming Xiong, Dragomir Radev
TACL 2022
[Paper] [Code]

Investigating Crowdsourcing Protocols for Evaluating the Factual Consistency of Summaries
Xiangru Tang, Alexander Fabbri, Haoran Li, Ziming Mao, Griffin Adams, Borui Wang, Asli Celikyilmaz, Yashar Mehdad, Dragomir Radev
NAACL 2022
[Paper]

Enhancing fouling mitigation of submerged flat-sheet membranes by vibrating 3D-spacers
Yong Zen Tan, Ziming Mao, Yanjun Zhang, Wen See Tan, Tzyy Haur Chong, Bing Wu, Jia Wei Chew
Sep. Purif. Technol. 2019
[Paper]

Spacer vibration for fouling control of submerged flat sheet membranes
Bing Wu, Yanjun Zhang, Ziming Mao, Wen See Tan, Yong Zen Tan, Jia Wei Chew, Tzyy Haur Chong, Anthony G Fane
Sep. Purif. Technol. 2019
[Paper]

Miscellaneous

  • I grew up in Ningbo, China and studied in Singapore at Hwa Chong Institution.
  • I play badminton every week! 🏸
  • I will be down for any stargazing trips 🔭 :)