Staff Machine Learning Engineer (AI)

Location: Menlo Park, California, United States. Compensation is assigned according to the U.S. work location region.

Salary: $232,000–$348,000 USD annually for Region A, including Menlo Park.

Company

AI advertising technology company building machine learning solutions for mobile user acquisition, retail media, marketplace monetization, and digital commerce growth.

What you will do

• Design, build, and productionize ML models for CTR prediction, retrieval, conversion, search, ranking, and recommendations.
• Lead experimentation from problem definition and offline modeling through online A/B testing and iterative optimization.
• Translate retail media and ecommerce requirements into ML solutions, features, metrics, and evaluation frameworks.
• Improve ML pipelines, model serving, monitoring, scalability, latency, throughput, and inference reliability.
• Own high-impact technical initiatives and influence architecture and roadmap decisions as the senior technical individual contributor.
• Mentor machine learning engineers through design reviews, knowledge sharing, and engineering best practices.

Requirements

• 8–12 years of experience in applied machine learning or machine learning engineering.
• Track record of shipping production models for advertising, recommender systems, search, or large-scale ecommerce.
• Hands-on expertise with Python, PyTorch, TensorFlow or JAX, feature stores, distributed training, and large-scale data processing.
• Experience building CTR predictors, retrieval models, conversion models, or recommendation systems; transformer-based architectures are relevant.
• Ability to define metrics, structure ambiguous problems, run experiments, and make data-driven iterations.
• Ability to lead technical initiatives, collaborate cross-functionally, and communicate complex ML concepts to non-technical stakeholders.

Nice to have

• Experience with transformer-based architectures.
• Background in retail media, ecommerce, advertising auctions, or marketplace products.

Culture & Benefits

• Medical, dental, and vision insurance for U.S.-based employees.
• 401(k) plan with company match, disability coverage, life insurance, and well-being benefits.
• Up to 12 scheduled paid holidays and one quarterly Thrive Day for U.S.-based employees.
• Flexible Time Off for all employees.
• Potential eligibility for performance-based bonuses and equity awards.
• Culture emphasizing humility, continuous growth, measurable customer value, inclusion, and collaboration.

Hiring process

• Interview process includes rounds designed to assess relevant skills and experience.
• Unless a round explicitly assesses AI use, candidates are expected to participate without AI assistance.