Senior Staff Software Engineer, AI Agent Platform, Vertex

About the job

As a part of the Cloud AI Agent Platform team (previously known as Vertex), you will focus on building highly differentiated, highly scalable, and easy-to-use GenAI products and services that enable customers to transform their business with GenAI.

You will be responsible for building the high-performance, hyper-scale backend infrastructure that powers the Vertex AI APIs, serving millions of requests and enabling the next-generation of AI-driven applications.

In this role, you will have a unique opportunity to be at the absolute forefront of the Generative AI revolution. You will be building the foundational API layer and infrastructure that serves massive foundation models (like Gemini) to the world. In this role, you will address unprecedented scaling issues, optimize for ultra-low latency, and design scalable systems that define how developers interact with AI.

You will operate in a highly dynamic, incredibly changing environment where your work will have an immediate and massive impact on Google Cloud's most product area.

Poland: zł640000 - zł655000 (PLN) + 25% bonus target + equity + benefits

Learn more about benefits at Google.

Responsibilities

• Architect, develop, and maintain horizontally scalable APIs and backend infrastructure tailored specifically for Generative AI workloads.
• Optimize the routing, load balancing, quota management, and data plane mechanisms that connect user API requests to backend ML serving clusters across massive fleets of TPUs and GPUs.
• Prototype and launch new API products rapidly in a highly collaborative environment, to expose the latest advancements in Google's foundation models.
• Collaborate closely with ML researchers, ML serving teams, and product managers to translate complex AI capabilities into intuitive, highly reliable enterprise-grade APIs.
• Define and implement best practices for API security, high availability, fault tolerance, and comprehensive observability across distributed systems.

Qualifications

Minimum qualifications:

• Bachelor’s degree or equivalent practical experience.
• 8 years of experience in software development.
• 7 years of experience leading technical project strategy, ML design, and working with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning).
• 5 years of experience in systems architecture and large-scale serving infrastructure.
• Experience delivering technical solutions in a changing environment.
• Experience working in multinational or internationalised (i18n) organisations on global initiatives.

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.
• 5 years of experience in a technical leadership role leading project teams and setting technical direction.
• 3 years of experience working in a complex, matrix organisations involving cross-functional, cross-timezone or cross-business projects.
• Experience in serving Generative AI models using inference frameworks or hardware accelerators (e.g., GPUs, TPUs).