Staff AI Performance Engineer

About the role

About the Job

Lead performance work that turns large-scale AI workloads into faster, more efficient systems.

As a Staff AI Performance Engineer, you will analyze and optimize AI training and inference workloads across distributed systems. Your work will connect compute, memory, networking and software behavior.

You will lead complex performance workstreams across hardware and software. You will find bottlenecks, validate improvements and turn performance evidence into practical engineering decisions.

You will build benchmarks, profile demanding workloads and develop C++ and Python tools for analysis and optimization. Your recommendations will help teams improve efficiency, scalability and reliability.

This role gives you broad technical scope across AI infrastructure at data center scale. It is based in Milpitas, California.

The team and culture

The System Engineering Performance team architect

Skills