Data Scientist II, ML Infrastructure
About the role
Pinterest is seeking a highly capable Data & Applied Scientist to drive step function improvements in our ML capabilities. This role focuses on advancing the science and systems behind ML measurement, feature understanding, and causal inference at scale.
Responsibilities
- Translate research-grade DS workflows into production ML pipelines using Airflow, WandB & Ray
- Apply and productionize causal inference methods using the production ML stack (propensity scoring, IPW, TMLE)
- Build self-serve tooling to empower non-experts to derive rigorous causal insights at scale
- Partner with ML engineers and product teams to identify opportunities for improved tooling, metrics, and measurement methods
- Leverage Pinterest's rich metadata and engagement signals to build data-driven frameworks, from feature importance to content deindexing
- Design and build centralized ML platform tooling to improve feature and model creation, evaluation, and trust
Requirements
- 2+ years of hands-on experience as an applied scientist, ML engineer, research scientist or software engineer, with significant ML production experience
- Strong Python skills; experience with PyTorch or equivalent deep learning frameworks; familiarity with distributed compute (Spark, Ray)
- Ray specifically is a strong plus
- Enthusiasm for building tools and platforms that multiply the impact of an entire ML organization
- Deep ML theory knowledge with extremely strong fundamentals that can help us reason about ML models from first principles
- Proficiency in software development best practices including version control, code review, and reproducible ML pipelines
- Experience with workflow management tools (Airflow, Prefect, Jenkins, or similar) for reliable ML pipeline orchestration
Qualifications
- Bachelor’s/Master’s degree in a relevant field such as Computer Science, or equivalent experience
Skills
- Experience with machine learning and data science
- Experience with distributed computing frameworks
- Familiarity with causal inference methods
- Experience with production ML systems
Benefits
At Pinterest, we offer a competitive compensation package, including a base salary range of $114,297—$235,319 USD, and eligibility for equity. We also provide comprehensive benefits, including health insurance, retirement plans, and paid time off.
Pay
$114,297—$235,319 USD
Schedule
This role requires in-office collaboration 3-5 times/quarter.
Relocation Statement
This position is not eligible for relocation assistance.
In-Office Requirement Statement
This role will need to be in the office for in-person collaboration 3-5 times/quarter and therefore can be situated anywhere in the country.