About the company
Millions of people around the world come to Pinterest to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we’re on a mission to bring everyone the inspiration to create a life they love.
About the role
Pinterest’s Product ML Infrastructure (PMLI) team enables fast, safe, and efficient delivery of AI/ML solutions across Ads and Core critical products. This role will set technical direction across model training and serving, with a focus on GPU efficiency and large-scale ranking systems.
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.
- Partner with Ads and Core AI/ML teams to productionize features and models.
- Drive cross-organizational architecture decisions, migrations, and operational standards.
- Use AI-assisted development and analysis.
Requirements
- A track record of setting technical strategy and delivering company-wide infrastructure initiatives.
- Deep expertise in distributed ML systems, including production experience with large-scale training and online inference.
- Strong GPU performance knowledge (profiling, distributed execution, kernel and memory optimization, compilation, quantization).
- Experience with AI/ML modeling, recommender systems, Ads ranking, retrieval, feature platforms.
- Strong systems programming and design skills in C++, Java, or Python.
- Bachelor’s degree in Computer Science, Engineering, or equivalent experience.
Conditions
- This position is not eligible for relocation assistance.
- In-office requirement: 1–2 times per quarter, within commutable distance of Palo Alto or San Francisco office.
- US based applicants only.