Google

Senior Software Engineer, AI/ML, ML Training

Google
Apply
1 day ago
Kirkland, WA, USA or Sunnyvale, CA, USASenior
H1B sponsor

Base Salary

$174k - $252k/yr

Responsibilities

  • Write and test product or system development code.
  • Collaborate with peers and stakeholders through design and code reviews to promote coding best practices, accuracy, testability, and efficiency.
  • Contribute to and update documentation or educational content based on product updates and user feedback.
  • Triage, debug, track, and resolve product or system issues by analyzing impacts on hardware, networks, service operations, and quality.
  • Design and implement solutions in specialized machine learning areas while leveraging ML infrastructure.

Requirements

  • Bachelor’s degree or equivalent practical experience.
  • At least 5 years of software development experience in one or more programming languages.
  • At least 3 years of experience testing, maintaining, or launching software products and 1 year of software design and architecture experience.
  • At least 3 years of end-to-end machine learning experience, including model deployment, model evaluation, data processing, and debugging.
  • At least 3 years of generative AI or performance analysis experience.
  • Preferred: master’s degree or PhD in computer science or a related technical field.
  • Preferred: 5 years of data structures and algorithms experience.
  • Preferred: 2 years of experience with ML accelerators such as TPUs and GPUs and their performance tuning.
  • Preferred: 2 years of experience optimizing Mixture of Experts or diffusion model performance.
  • Experience building and releasing software artifacts such as pip wheels and Docker images.

Benefits

  • US base salary of $174000 - $252000, plus a 15% bonus target, equity, and benefits.
  • Work focuses on Google Cloud AI Hypercomputer workloads using TPUs, GPUs, and CPUs.
  • Opportunity to collaborate with DeepMind and Core ML teams and contribute to open-source implementations.
Google

About Google

10,000+ employees

Google builds consumer and enterprise software and services including Search, Android, YouTube, Chrome, Maps, Gmail, and Google Cloud. Its business model centers on digital advertising and paid cloud, software, and hardware offerings (e.g., Pixel and Nest) for consumers, developers, and organizations. Founded in 1998 and headquartered in Mountain View, California, Google operates globally as a subsidiary of Alphabet Inc.

Contact me