The Sr. Staff Software Engineer, Product ML Infrastructure role at Pinterest focuses on setting the technical vision and roadmap for model training and serving across the Product ML Infrastructure team. This team is dedicated to enabling fast, safe, and efficient delivery of AI/ML solutions across critical products. The position emphasizes improving GPU efficiency and large-scale ranking systems while collaborating with Ads and Core AI/ML teams. Candidates will lead architectural decisions and mentor senior engineers to enhance the engineering standards within the organization.

• A track record of setting technical strategy and delivering company-wide infrastructure initiatives in ambiguous environments.
• Deep expertise in distributed ML systems, including production experience with both large-scale training and online inference.
• Strong GPU performance knowledge, such as profiling, distributed execution, kernel and memory optimization, compilation, or quantization.
• Experience with AI/ML modeling, recommender systems, Ads ranking, retrieval, feature platforms, or similarly demanding ML workloads.
• Strong systems programming and design skills in C++, Java, or Python.
• High ownership and sound judgment in reliability, security, cost, and operational excellence.
• Demonstrated ability to use AI to improve speed and critically evaluate AI-assisted work, with accountability for correctness, quality, and sensitive data.
• Bachelor’s degree in Computer Science, Engineering, a related field, or equivalent experience.

• 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.