Staff Machine Learning Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Moloco · 1 day ago

Staff Machine Learning Engineer

Moloco builds powerful AI advertising solutions and is seeking a Staff Machine Learning Engineer to join their core ML team. This role focuses on building ML systems at massive scale that directly influence business outcomes, collaborating closely with infrastructure and product teams.

AdvertisingCloud InfrastructureMobileMobile Apps
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Design and deploy ML systems for recommendation, prediction, ranking, and optimization (e.g., clicks, conversions, user satisfaction)
Build and operate models and infrastructure that support hundreds of millions of requests per day
Apply systematic, reproducible approaches to model training, evaluation, and inference at scale
Collaborate with the Machine Learning Infrastructure (MLI) team to define system requirements and improve developer tooling and workflows
Lead or contribute to architectural design, technical reviews, and standardization of ML practices across the org
Mentor and support engineers across ML teams as a senior technical expert
Help evolve the ML interview process to better identify high-caliber candidates without compromising our technical bar

Qualification

ML systems designRecommender systemsPredictive modelingML optimizationModern ML frameworksDistributed infrastructureModel optimizationSystem design skillsCoding skillsCollaborationMentoring

Required

Extensive experience building and operating ML systems in high-scale environments
Demonstrated impact from ML systems that directly improved product performance or business metrics
Expertise in at least one of the following: recommender systems, ranking models, predictive modeling, or ML optimization
Strong coding and system design skills, with solid CS fundamentals
Proficiency with modern ML frameworks such as PyTorch, Keras, or JAX
Experience optimizing models for hardware accelerators (e.g. GPUs, TPUs)
Hands-on experience with distributed infrastructure – ideally as a contributor to frameworks like TensorFlow, Apache Beam, Dataflow, Flume, or Parquet
Familiarity with best practices in ML production, including data versioning, feature engineering, training pipelines, and model monitoring

Benefits

We offer innovative benefits that empower our employees to take care of themselves and their families so they can do the best work of their lives.

Company

Moloco offers programmatic advertising solutions to help optimize the client's acquisition, retention, and monetization campaigns.

H1B Sponsorship

Moloco has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (50)
2024 (38)
2023 (29)
2022 (11)
2021 (4)
2020 (13)

Funding

Current Stage
Late Stage
Total Funding
$215.75M
Key Investors
Premier PartnersTiger Global ManagementLG Technology Ventures
2023-06-29Series D· $24.18M
2023-06-29Secondary Market
2021-08-17Series C· $150M

Leadership Team

leader-logo
Ikkjin Ahn
Co-founder and CEO
linkedin
leader-logo
David Sehyuk Park
Co-founder/CIO
linkedin
Company data provided by crunchbase