Quick overview
Job description
About the Role
Join a small, early-stage AI team building a recommendation system for personalized advertising. You will own the matching stack end to end, from data and models through real-time serving, and help shape the system's technical priorities as it scales.
What You'll Do
Build retrieval, ranking, and re-ranking systems to match ads with recipients.
Develop embedding and vector-search infrastructure for real-time ad selection.
Optimize ranking across click-through rate, advertiser return, publisher revenue, and subscriber relevance.
Own the feature store, offline training, online inference, serving path, and pipeline health.
Apply large language models and foundation models to subscriber tagging, content understanding, and user modeling.
Take initiatives from prototype to production, including online learning, feedback loops, ETL/ELT pipelines, and warehouse transformations.
Create dashboards and automated scorecards to track performance and key metrics.
Partner with company leadership and engineering to set priorities, with an opportunity to grow into leading an ML team.
What We're Looking For
At least 3 years of experience with recommendation, ranking, or advertising systems, and 5 to 16 years of overall experience.
Production experience owning recommendation or matching systems across candidate generation, ranking, embedding-based retrieval, serving, and evaluation.
Experience building ad ranking, click-through or conversion prediction, or ad-matching models.
Hands-on experience with embeddings and real-time vector search tools such as FAISS, Milvus, Pinecone, pgvector, or similar.
Strong SQL skills and experience building data pipelines and transformation layers with dbt or similar tools, plus Postgres or columnar analytical databases.
Early-stage startup or founder experience is valued; experience building an ML team and graduate study in computer science or machine learning are also relevant.
Compensation & Benefits
Base salary: $200,000 to $300,000 annually. Visa sponsorship is not available.
Location
Hybrid, with preference for candidates in the New York City area or in a nearby U.S. time zone who can travel to New York City quarterly.
Similar jobs
- CO

Staff Machine Learning Engineer, Offsite AdsRecently posted Auto Apply
Coupang · Mountain View · Hybrid 🇺🇸
MLflowScikit-learnAirflow+9Technology$164k - $282k/yr - CO

Senior Machine Learning EngineerRecently posted Auto Apply
Coinbase · Remote - USA · Remote 🇺🇸
A/B TestingArbitrationData Privacy+8Technology$213.8k/yr - WA

2027 Summer Intern, PhD, Machine Learning - Generative World Models & Controllable SimulationRecently posted Auto Apply
Waymo · Mountain View · Hybrid 🇺🇸
RoboticsComputer VisionDeep Learning+4Technology$85/hr - WA

2027 Summer Intern, Machine Learning, MS/PhD, Simulation Realism (World Model)Recently posted Auto Apply
Waymo · Mountain View · Hybrid 🇺🇸
RoboticsComputer VisionDeep Learning+4Technology$70 - $85/hr - CL

Senior ML Engineer, PerceptionRecently posted Auto Apply
Clera · San Francisco · On-site 🇺🇸
RoboticsComputer VisionC+++1Technology$190k - $222k/yr - RM

Machine Learning Scientist II, Drug Discovery AnalyticsRecently posted Auto Apply
Revolution Medicines · Redwood City · Hybrid 🇺🇸
MLOpsNumPySciPy+9Technology$202k - $238k/yr - TR
Robotics Intern, World Action Models for Robot PolicyRecently posted Auto Apply
Tri · Cambridge · Hybrid 🇺🇸
RoboticsDeep LearningC+++3Salary not listed - SE
Staff Machine Learning Engineer - VoIP Infrastructure Auto Apply
ServiceNow · Dallas, Texas · Remote 🇺🇸
GitSplunkAnsible+10Technology$176.1k - $308.2k/yr - SE
Staff Machine Learning Engineer - VoIP Infrastructure Auto Apply
ServiceNow · Seattle, Washington · Hybrid 🇺🇸
GitSplunkAnsible+10Technology$176.1k - $308.2k/yr - SE
Staff Machine Learning Engineer - VoIP Infrastructure Auto Apply
ServiceNow · Boston, Massachusetts · Remote 🇺🇸
GitSplunkAnsible+10TechnologySalary not listed - SE
Staff Machine Learning Engineer - VoIP Infrastructure Auto Apply
ServiceNow · Santa Clara, CALIFORNIA · On-site 🇺🇸
GitSplunkAnsible+10Technology$176.1k - $308.2k/yr - RU
Principal Engineer, Model PerformanceRecently posted Auto Apply
Runware · United States · Remote 🇺🇸
Load BalancingCUDADeep Learning+5TechnologySalary not listed