{bc}
Apply
company_site

Staff Software Engineer, AI/ML

Google
Kirkland, USA
Senior · 8+ years experience
Onsite
USD 207000-300000 / year
Discovered Today

Full Job Posting

Responsibilities

  • Help make MaxText the best pre and post training solution for TPUs on GCP.
  • Implement LLM modeling and scaling techniques in MaxText, including multi-modal, Physical AI and World models.
  • Build and support JAX and its ecosystem to benefit GCP customers.
  • Support new accelerator introductions (TPUs/GPUs) on GCP and MLPerf submissions that demonstrate cutting edge technologies for Gen AI workloads at scale.
  • Work closely with product managers and partner teams to deliver high-quality software products and features throughout their lifecycle.

Minimum qualifications:

  • Bachelor’s degree or equivalent practical experience.
  • 8 years of experience in software development.
  • 2 years of experience with Applied Machine Learning (ML), including the development and deployment of models.
  • Experience with Large Language Models (LLMs), NLP, or Generative AI.

Preferred qualifications:

  • Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
  • 8 years of experience with data structures and algorithms.
  • 3 years of experience in a technical leadership role leading project teams and setting technical direction.
  • Experience with performance optimizations for machine learning workloads running on CPU/GPU/TPU.

Apply for this job in 1 click

Skip the repetitive application forms

Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.

Sarah M.James T.Maya R.

Trusted by over 500,000 job seekers on Base Career

Start Free Today

More from this employer

More jobs at Google

Technical Program Manager III, Workspace Delivery, Cloud Networking

Austin, USA

Discovered Today

Manager, Network Implementation Engineering

Midlothian, USA

Discovered Today

Director, 1P Direct to Consumer Commerce and Performance Marketing

San Francisco, USA

Discovered Today

Game Designer, Inception, DeepMind (Fixed-Term Contract)

London, GBR

Discovered Today

Senior Financial Analyst, Technical Infrastructure

Sunnyvale, USA

Discovered Today

Tech Partner Lead, Google Marketing Platform (Fixed Term Contract)

Sydney, AUS

Discovered Today

Senior Staff Software Engineer, Google Cloud Business Platform

Sunnyvale, USA

Discovered Today

Staff Software Engineer, BigQuery Processing Core Runtime

Kirkland, USA

Discovered Today