Senior Software Engineer, Machine Learning

About the Role

As a Senior Software Engineer on the Machine Learning team, you will be responsible for designing, building, and deploying scalable ML models that power Google's core products. You will work closely with research scientists to bring state-of-the-art algorithms into production environments, directly impacting billions of users.

We are looking for someone who is deeply passionate about performance optimization, distributed systems, and modern AI architectures. You will lead technical initiatives, mentor junior engineers, and set the standard for ML engineering excellence within the org.

Key Responsibilities

• check_circle Design and implement high-performance, distributed machine learning training and inference pipelines.
• check_circle Optimize model serving infrastructure for latency and throughput across global data centers.
• check_circle Collaborate cross-functionally with product managers, data scientists, and infrastructure engineers.
• check_circle Write clean, maintainable, and rigorously tested C++ and Python code.

Minimum Qualifications

• Bachelor's degree in Computer Science, related technical field, or equivalent practical experience.
• 5+ years of software development experience using Python, C++, or Go.
• Experience with ML frameworks such as TensorFlow, PyTorch, or JAX.
• Experience building and operating large-scale distributed systems.

Preferred Qualifications

• Master's degree or PhD in Computer Science or Artificial Intelligence.
• Experience optimizing deep learning models for specific hardware accelerators (TPU, GPU).
• Contributions to open-source machine learning projects.