Wei Bai 白巍
Building the networking systems that large-scale AI runs on — reliably, in production.
About
I am a Member of Technical Staff at Microsoft AI, where I focus on networking for large-scale AI clusters. Previously, I was a Principal Software Research Architect at NVIDIA and a Senior Researcher at Microsoft Research Redmond. I received my Ph.D. from Hong Kong University of Science and Technology and my B.E. from Shanghai Jiao Tong University.
I have worked on computer networking for over a decade, with a long focus on data center networks and, more recently, the networks behind large-scale AI training and inference. My work spans RDMA, Ethernet transport, congestion control, and production reliability across research prototypes, hardware/software co-design, and hyperscale deployments.
I still enjoy stress-testing NICs and switches until the weak points become clear. Beyond networking, I am increasingly interested in how AI agents can improve engineering workflows and day-to-day productivity.
Outside of work, I enjoy life with my love Zhen Lin between Seattle and Vancouver.
News
Recent Professional Activities
Selected Publications
See my full list of publications →