AI/ML Infrastructure Software Engineer
Back to JobsGoogle Get Smart Job AI Coach in the appFree on iOS and Android Google
Google
Google
Google
Google
Google
AI/ML Infrastructure Software Engineer
Location
Bengaluru, Karnataka, India
Experience
Senior
Posted
Jul 18, 2026
Apply by
August 17, 2026
Applicants
0
Early applicantEasy applyFull-timeWork from Office
Job Description
Minimum qualifications: Bachelor's degree or equivalent practical experience. 8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript). Experience in ML Infrastructure Development or ML Performance Engineering. Preferred qualifications: Experience with Machine Learning (ML) development process. Expertise in C++ and Python, Responsible ML, ML Explainability, ML Fairness Tensorflow , JAX, Keras, Gemax TFX, TFlex, Xflow or ML Performance Engineering. Knowledge of Google production infrastructure. Ability to collaborate with partners on applying ML to their use cases. Passion for building infrastructure to increase the velocity of ML development.
About the job The AI Flow Platform and suite of products are used by thousands of Googlers across all product areas to drive business impact for Google in innumerable ways (e.g., finding insights in data, building classical ML models, experimenting with Gemini, and more). Our team provides a dependable, compliant, and scalable platform that enables product teams to iterate rapidly on new ideas and to take the best ideas for production. In this role, you will have an opportunity to influence the future of applied ML at Google, pushing the boundaries of what teams can achieve with generative and discriminative technologies, leaving a lasting impact on the field.
Responsibilities Develop and maintain our production service, written in C++ and Python, and further the platform goal of incorporating prompt optimization, fine tuning, hyperparameter optimization, and ML best practices into Google product development. Design and implement customer requested ML and AI features end-to-end from requirements-gathering and design-authoring, through implementation and landing impact in production. Explore charter expansion based on next-generation product requirements from internal and external customers.
Key Responsibilities
- Develop and maintain production services in C++ and Python.
- Incorporate prompt optimization, fine-tuning, and hyperparameter optimization into product development.
- Design and implement customer-requested ML and AI features end-to-end.
- Explore charter expansion based on next-generation product requirements.
Requirements
- Bachelor's degree or equivalent practical experience
Skills Required
PythonC++CJavaJavaScriptML Infrastructure DevelopmentML Performance EngineeringCollaborationMachine Learning development processTensorflowJAXKerasGemax TFXTFlexXflowGoogle production infrastructurePassion for building infrastructure
App exclusive · Free
Smart Job AI Coach
Your personal interview coach on every job — readiness tips, profile improvements, and role-specific prep. Available only in the Pulse Job app.
Interview readiness
See how prepared you are and what to improve for each role.
Personalized tips
Actionable suggestions based on your profile and the job.
After you apply
Keep coaching momentum from job detail through application success.
Similar roles for you
Matched using this role's title and skills. Open the job search anytime to see every listing.
Software Engineer III, AI/ML, gUP Customer Support
147,000–211,000 / Year
Full-timeEasy applyWork from Office
SunnyvaleMid
Senior Software Engineer, Generative AI, gUP
174,000–253,000 / Year
Full-timeEasy applyWork from Office
SunnyvaleSenior
Software Engineer III, AI/ML Shopping Creator, YouTube
Full-timeEasy applyWork from Office
ZürichMid
Software Engineer, Content Safety
Full-timeEasy applyWork from Office
SingaporeMid
Staff Software Engineer, AI/ML Infrastructure, TPU Supercomputers
207,000–301,000 / Year
Full-timeEasy applyWork from Office
SunnyvaleSenior
Senior Software Engineer, AI/ML GenAI, Google Cloud
174,000–253,000 / Year
Full-timeEasy applyWork from Office
SunnyvaleSenior