Senior Staff ML Infra Engineer - Product Platform

Pinterest; Palo Alto (CA); Hybrid; USD 245,000 - 429,000; Full time

Job summary

Pinterest is seeking a Sr. Staff Software Engineer for Product ML Infrastructure to set the technical vision for model training and serving.

You will lead distributed training architectures and collaborate with Ads and Core AI/ML teams to deploy scalable features at Pinterest scale. The role emphasizes ownership, reliability, and the use of AI to accelerate prototyping while ensuring correctness and data safeguards.

Qualifications

• Track record of setting technical strategy and delivering infrastructure initiatives.
• Experience with distributed ML systems and large-scale training.
• Strong GPU performance knowledge (profiling, kernel optimization).
• Experience with AI/ML modeling, recommender systems, Ads ranking or similar workloads.

Responsibilities

• Set the technical vision and roadmap for model training and serving across PMLI.
• 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.

Skills

Distributed ML systems

C++

Java

Python

GPU optimization

System design

Reliability

Education

Bachelor's degree in CS/Engineering or related field

Job description

Pinterest is seeking a Sr. Staff Software Engineer for Product ML Infrastructure to set the technical vision for model training and serving.

You will lead distributed training architectures and collaborate with Ads and Core AI/ML teams to deploy scalable features at Pinterest scale. The role emphasizes ownership, reliability, and the use of AI to accelerate prototyping while ensuring correctness and data safeguards.