Manfei Bai - Software Engineer III at Google

Manfei Bai

Software Engineer III at Google

Boston, Massachusetts, United States

Join Prog.AI to see contacts

Summary

🤩

Rockstar

🎓

Top School

Manfei Bai is a software engineer with seven years of experience building backend systems, currently a Software Engineer III at Google in Boston. She has a strong background in low-level ML infrastructure, contributing to the high-profile pytorch/xla project where she implemented XLA codegen for ops like atan2 and baddbmm and helped harden test/runtime integration. Her experience includes multiple internships and engineering roles at Intel and progressive responsibility at Google, reflecting a focus on performance-sensitive systems and cross-team maintenance. Holding a master's from Northeastern and a CS bachelor's from Southeast University, she combines academic rigor with practical production engineering. An understated strength is her track record of improving test stability and tooling—which often yields outsized productivity gains in large ML codebases.

7 years of coding experience

2 years of employment as a software developer

Bachelor's degree, Computer Science and Software Engineering, Bachelor's degree, Computer Science and Software Engineering at Southeast University

Master's degree, Master's degree at Northeastern University

Github Skills (9)

xla10

compiler10

pytorch10

deeplearning-ai10

compiler-compiler10

deep-learning10

cprogramming-language9

c-language9

tensorflow8

Programming languages (6)

C++ShellScalaJupyter NotebookPythonJsonnet

Github contributions (5)

pytorch/xla

Nov 2022 - Jan 2023

Enabling PyTorch on XLA Devices (e.g. Google TPU)

Role in this project:

Back-end Developer

Contributions:4 releases, 666 reviews, 66 commits in 2 months

Contributions summary:Manfei primarily contributed to the codebase by implementing and modifying functionalities related to the XLA backend for PyTorch. This included the addition of codegen for `atan2` and `baddbmm` operations, enhancing the XLA support for the PyTorch framework. Furthermore, the user worked on disabling failing PyTorch tests and updating TF pins. These changes reflect a focus on extending and maintaining the core functionality of the PyTorch/XLA integration.

pytorchxladeep-learningtpucompiler

ManfeiBai/ml-testing-accelerators

Dec 2022 - Dec 2023

Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)

Contributions:2 PRs, 89 pushes, 6 branches in 11 months

pytorchtesting-frameworkdeep-learninggpuhardware-accelerators

Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.

Request Free Trial