PhD AI/ML Systems Engineer — Equity & Bonus

Socket.dev; Sunnyvale (CA); On-site; USD 147,000 - 210,000; Full time

Job summary

Google is seeking a PhD Software Engineer in Sunnyvale to push the boundaries of ML/AI at scale. You will collaborate with world-class researchers and engineers across products spanning AI, infrastructure, YouTube, Search, Ads, and more.

You will lead design and implementation of end-to-end ML systems, from hardware acceleration to production APIs, applying your research to real-world applications while growing your expertise in a diverse, innovation-driven environment.

Qualifications

• PhD in CS/AI/ML or equivalent practical experience.
• Proficient in Python, C, C++, Java, JavaScript or Go.
• Experience in ML/AI.

Responsibilities

• Collaborate on design, analysis, and development of advanced ML systems across the stack.
• Support end-to-end ML systems from hardware acceleration to production APIs.
• Optimize performance, memory usage, and reliability in production.
• Write clear, tested code and participate in code reviews to uphold quality.

Skills

ML/AI expertise

Research mindset

Team collaboration

Education

PhD in Computer Science / AI / ML or related field

Tools

Python

C

C++

Java

JavaScript

Golang

Rust

PyTorch

TensorFlow

JAX

Job description

Google is seeking a PhD Software Engineer in Sunnyvale to push the boundaries of ML/AI at scale. You will collaborate with world-class researchers and engineers across products spanning AI, infrastructure, YouTube, Search, Ads, and more.

You will lead design and implementation of end-to-end ML systems, from hardware acceleration to production APIs, applying your research to real-world applications while growing your expertise in a diverse, innovation-driven environment.