Moloco

AI advertising solutions

MachineLearningEngineer,Infrastructure

$167–210k Menlo Park, California, United States; San Francisco, California, United States Remote Friendly
Market Sentiment
HIGH DEMAND

Neural analysis suggests this role is
optimal for Mid candidates.

The Brief

“Machine Learning Engineer, Infrastructure at Moloco. Skills: ML infrastructure, JAX, Rust, Python. Drive development and optimization of ML infrastructure. Enhance performance and reliability of Ads product line”

What You'll Achieve.

Deliver high-quality, data-driven advertising solutions at scale; Deliver end-to-end ML solutions; Achieve vision to become scaling engine for Internet economy

Industry & Context.

AI advertising solutions
Problems you'll solve

Proactively identifying and solving complex infrastructure challenges; Apply problem-solving skills

What They're Looking For.

Must Have

4+ years of experience in machine learning engineering, Proficiency in programming languages such as Python, Java, C++, or Rust, Hands-on experience with ML frameworks and libraries (e. g. , TensorFlow, PyTorch, Keras, Jax), Work with cloud platforms (e. g. , AWS, GCP, Azure) and containerization/orchestration tools (e. g. , Docker, Kubernetes), Build and maintain scalable data pipelines using tools like Apache Beam, Apache Spark, Airflow, or similar, Thrive in ambiguous environments, Collaborate effectively with cross-functional teams, Apply problem-solving skills and a growth mindset

Nice to Have

Experience with JAX and Rust, Experience with GPUs and TPUs

What You'll Do.

Drive development and optimization of ML infrastructure

Enhance performance and reliability of Ads product line

Empower teams to deliver ML solutions

and maintain ML infrastructure

Support large-scale ad serving

Support model training at global scale

Develop and optimize data pipelines

Optimize workflows for model deployment

Optimize workflows for model monitoring

Implement best practices for model versioning

Implement best practices for reproducibility

Implement best practices for CI/CD

Build and operate high performance ML systems

Optimize execution on GPUs

Optimize execution on TPUs

Troubleshoot ML systems

Improve reliability of ML systems

Improve scalability of ML systems

Improve performance of ML systems

Evaluate and integrate new tools

Evaluate and integrate new frameworks

Evaluate and integrate new technologies

Enhance ML platform capabilities

Integrate AI-driven agents

Automate engineering lifecycle

Contribute to design of experiments

Contribute to execution of experiments

Improve system performance

Document system architecture

Document best practices

How You'll Work.

Team & Collaboration

Collaborate with cross-functional teams—including data scientists, product managers, and software engineers; Collaborate proactively and inclusively; Involve the right people at the right time

Full Job Description

About Moloco: Moloco builds some of the most powerful AI advertising solutions in the world. Our name—short for "machine learning company"—reflects our core mission: democratizing access to the advanced AI that has historically been reserved for tech giants. Led by machine learning pioneers who built some of the most successful ad systems at Google, including YouTube's monetization engine and key search advertising technologies, we're transforming how businesses grow and compete in the digital economy. Built with AI from day one, Moloco’s planet-scale machine learning platform powers a suite of solutions for advertising growth and monetization. Moloco Ads is an AI-powered platform that delivers real business outcomes for mobile app marketers through performance-based user acquisition. Moloco Commerce Media enables retailers and marketplaces to build revenue-generating ad businesses that balance user experience and advertiser performance. Moloco is headquartered in Silicon Valley, with offices in Seattle, New York, San Francisco, Seoul, Beijing, Singapore, Gurgaon, Tokyo, Shanghai, London, Tel Aviv, and Berlin. Moloco is a truly rewarding place to work and in an exciting period of growth, which you could be a part of. Join us today and apply now! The Impact You’ll Be Contributing to Moloco As a Machine Learning Engineer focused on ML Infra, you will drive the development and optimization of scalable machine learning infrastructure, directly enhancing the performance and reliability of our Ads product line. Your work will empower teams to deliver high-quality, data-driven advertising solutions at scale. The Opportunity Design, build, and maintain robust machine learning infrastructure to support large-scale ad serving and model training at a global scale. Develop and optimize data pipelines and workflows for efficient model deployment and monitoring. Collaborate with cross-functional teams—including data scientists, product managers, and software engineers—to deliver

Free ATS check

Applying for this Machine Learning Engineer, Infrastructure role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

How to Apply on Greenhouse

  • Create a Greenhouse profile before applying — it saves time across multiple applications.
  • Upload your resume as a PDF; the parser handles it better than Word.
  • Answer all knockout questions carefully — wrong answers auto-reject before a human sees you.
  • Enable email notifications to track application status in real time.

ANONYMOUS · UNFILTERED

What do employees actually say about Moloco?

Real rants from real employees. Read before you apply.

Read Company Rants →