PhD AI/ML Systems Engineer — From Research to Production

Google; United States; Hybrid; USD 147,000 - 210,000; Full time

Benefits offered by this job

Health insurance

401(k) with company match

Paid Time Off

Job summary

Google is seeking a PhD Software Engineer specialized in AI/ML to advance next-generation ML systems at scale. You will apply deep research to design, optimize, and deploy ML models across the stack, from hardware acceleration to production APIs, impacting multiple product areas.

Ideal candidates bring proven ML experience, strong coding in Python and C/C++, and familiarity with PyTorch, JAX, or TensorFlow.

Qualifications

• PhD in CS/AI/ML or equivalent practical experience.
• Experience coding in Python, C, C++, Java, JavaScript or Golang.
• Experience in Machine Learning or Artificial Intelligence.

Responsibilities

• Collaborate or lead on team projects to design, analyze, and develop advanced ML systems across the stack.
• Support building end-to-end ML systems from hardware acceleration to production APIs.
• Optimize complex system performance and fix bottlenecks and memory issues.
• Promote engineering quality through well-tested code and code reviews.

Skills

Python

C++

Java

Go

Rust

Machine Learning

ML Systems

Education

PhD in CS/AI/ML or related field

Tools

PyTorch

JAX

TensorFlow

Rust

Job description

Google is seeking a PhD Software Engineer specialized in AI/ML to advance next-generation ML systems at scale. You will apply deep research to design, optimize, and deploy ML models across the stack, from hardware acceleration to production APIs, impacting multiple product areas.

Ideal candidates bring proven ML experience, strong coding in Python and C/C++, and familiarity with PyTorch, JAX, or TensorFlow.