Apply on Employer Site

Google · 9 hours ago

Staff Software Engineer, AI/ML Agentic Migrations

Mountain View, CA

Full-time

Onsite

Lead/Staff

$197K/yr - $291K/yr

8+ years exp

Google is a leading technology company that develops next-generation technologies to change how billions of users connect and interact with information. The Staff Software Engineer will design, develop, test, and enhance software solutions, focusing on Machine Learning frameworks and collaboration with ML Infrastructure teams.

AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO

Growth Opportunities

H1B Sponsor Likely

Responsibilities

Design and development of related Machine Learning (ML) Frameworks Agentic solutions

Help to build a multi-agent system to deliver significant and measurable time savings on real-world migrations of ML models and their training pipelines

Work in collaboration with ML Infrastructure teams and Product Areas of ML Infrastructure (e.g., YouTube, Search)

Give contributions to the Artificial Intelligence (AI) for AI roadmap

Qualification

Machine Learning designGenerative AI techniquesML FrameworksPythonSoftware architectureSystem designing skillsLeadership qualitiesProject management

Required

Bachelor's degree or equivalent practical experience

8 years of experience in software development

5 years of experience in testing and launching software products, and 3 years of experience with software design and architecture

5 years of experience in leading Machine Learning (ML) design and optimizing ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning)

2 years of experience with Generative AI techniques (e.g., Large Language Models (LLMs), Multi-Modal, Large Vision Models) or with Generative AI-related concepts (e.g., language modeling, computer vision)

Preferred

Master's degree or PhD in Engineering, Computer Science, or a related technical field

Experience with coding in Python

Experience with ML model training and evaluation

Experience with ML Frameworks (e.g., Jax, Keras, TensorFlow, PyTorch), or machine learning production infrastructure

Experience with Agentic tools, prompt engineering, context management

Excellent system designing skills to understand the Machine Learning (ML) pipelines and take technical decisions in the greenfield project

Benefits

Bonus

Equity

Benefits

Company

Google

Glassdoor4.4

Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.

Founded in 1998

Mountain View, California, USA

10001+ employees

https://www.google.com

H1B Sponsorship

Google has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (8763)

2024 (8872)

2023 (9682)

2022 (11626)

2021 (9109)

2020 (9785)

Funding

Current Stage

Public Company

Total Funding

$26.1M

Key Investors

Andy Bechtolsheim

2004-08-19IPO

1999-06-07Series Unknown· $25M

1998-11-01Angel· $1M

Leadership Team

Sundar Pichai

CEO

Thomas Kurian

CEO - Google Cloud

Recent News

The Indian Express

Who is Ragini Das? Rejected by Google 12 years ago, she now leads its startup arm in India

2025-10-08

Tech.eu

Profits up at Google UK, but nearly 400 jobs cut

2025-10-04