G
Hiring companyGoogle
Verified today
AI/ML Infrastructure Software Engineer
Bengaluru,Karnataka,India Full-time On-site
About the role
AI Flow Platform team at Google builds a dependable, compliant, and scalable infrastructure that accelerates ML development across the company. The Software Engineer designs, implements, and maintains production services in C++ and Python, adding capabilities like prompt optimization, fine‑tuning, and hyperparameter tuning. Bengaluru, India, on-site.
What you’ll do
- Develop and maintain production services in C++ and Python
- Incorporate prompt optimization, fine‑tuning, hyperparameter optimization into platform
- Design and implement end‑to‑end ML and AI features for internal customers
- Gather requirements, author designs, and ship production impact
- Explore charter expansion based on next‑generation product needs
What you’ll bring
- Bachelor's degree or equivalent practical experience
- 8 years software development experience
- Proficiency in Python, C, C++, Java, or JavaScript
- Experience in ML infrastructure development or performance engineering
- Familiarity with TensorFlow, JAX, Keras, TFX or similar frameworks
- Knowledge of Google production infrastructure
- Ability to collaborate with cross‑functional partners
- Passion for building infrastructure to accelerate ML development
Skills
C++PythonML InfrastructureTensorFlowJAXKerasTFXProduction Systems