Sr. Staff Software Engineer, Product ML Infrastructure

📋 Description

• Set the technical vision and roadmap for model training and serving across PMLI, with reusable
• Lead architectures for distributed training, fine-tuning, distillation, evaluation, and high-scale
• Improve efficiency across data loading, distributed execution, GPU kernels and memory, compilation
• Build reliable, observable platforms with strong quality guarantees and training/serving
• Partner with Ads and Core AI/ML teams to productionize features and models safely at Pinterest
• Drive cross-organizational architecture decisions, migrations, and operational standards; mentor

🎯 Requirements

• A track record of setting technical strategy and delivering company-wide infrastructure initiatives
• Deep expertise in distributed ML systems, including production experience with both large-scale
• Strong GPU performance knowledge, such as profiling, distributed execution, kernel and memory
• Experience with AI/ML modeling, recommender systems, Ads ranking, retrieval, feature platforms, or
• Strong systems programming and design skills in C++, Java, or Python.
• High ownership and sound judgment in reliability, security, cost, and operational excellence.

🎁 Benefits

• Equity
• Position is not eligible for relocation assistance
• In-office requirement: 1–2 times per quarter within commutable distance of Palo Alto or San