Sr. Staff Software Engineer, Product ML Infrastructure

Pinterest is a platform that helps people discover creative ideas and plan meaningful experiences. The Product ML Infrastructure team is seeking a Sr. Staff Software Engineer to set technical direction for model training and serving, improve GPU and system efficiency, and build reliable AI/ML infrastructure at scale. The role partners with Ads and Core AI/ML teams, drives architecture and operational standards, and mentors senior engineers.

Responsibilities:

• Set the technical vision and roadmap for model training and serving across PMLI, with reusable interfaces to data and feature infrastructure
• Lead architectures for distributed training, fine-tuning, distillation, evaluation, and high-scale CPU/GPU inference
• Improve efficiency across data loading, distributed execution, GPU kernels and memory, compilation, quantization, scheduling, and capacity
• Build reliable, observable platforms with strong quality guarantees and training/serving consistency
• Partner with Ads and Core AI/ML teams to productionize features and models safely at Pinterest scale
• Drive cross-organizational architecture decisions, migrations, and operational standards; mentor senior engineers and raise the engineering bar
• Use AI-assisted development and analysis to accelerate prototyping, performance diagnosis, and validation while maintaining rigorous correctness and data safeguards