Senior Staff/Principal Engineer, Foundation Model Compute Infra

As a Senior Staff/Principal Engineer on the Foundation Model Compute Infrastructure team, you will architect, build, and operate large-scale infrastructure powering foundation model training, fine-tuning, evaluation, and inference. You will tackle challenging problems in distributed scheduling, accelerator utilization, cluster management, fault tolerance, and large-scale job orchestration. You will work with cloud and hardware partners to bring new TPU generations into production and build the tools and services that enable researchers to launch, monitor, debug, and profile their workloads without needing deep infrastructure expertise.

You'll develop model inference and fine-tuning services, build secure, scalable sandbox infrastructure for running AI agents, code execution, and reinforcement learning environments at massive scale, and work closely with foundation model researchers and engineers to improve performance, reliability, scalability, and developer productivity. This is a deeply hands-on technical leadership role. You will write, review, and debug production code, lead complex architectural decisions, and use modern AI-assisted development tools to accelerate engineering execution and raise the quality bar across the team.