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

Moloco · 8 hours ago

Machine Learning Engineer, Infrastructure

Moloco is a leading AI advertising solutions company dedicated to democratizing access to advanced machine learning technologies. As a Machine Learning Engineer focused on ML Infrastructure, you will enhance the performance and reliability of their Ads product line by developing and optimizing scalable machine learning infrastructure.

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

Responsibilities

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 end-to-end ML solutions
Implement best practices for model versioning, reproducibility, and continuous integration/continuous deployment (CI/CD) in ML systems
Build and operate high performance ML systems using modern frameworks and languages such as JAX and Rust, optimized for execution on GPUs and TPUs
Monitor, troubleshoot, and continuously improve the reliability, scalability, and performance of ML systems delivering millions of predictions per second worldwide
Evaluate and integrate new tools, frameworks, and technologies to enhance the ML platform’s capabilities
Integrate AI-driven agents into the core engineering and modeling lifecycle to automate and amplify the team's impact
Contribute to the design and execution of experiments to improve ad quality and system performance
Document system architecture, processes, and best practices to ensure knowledge sharing and maintainability

Qualification

Machine Learning EngineeringML InfrastructureData PipelinesProgramming LanguagesCloud PlatformsContainerization ToolsML FrameworksCollaborationProblem-SolvingGrowth Mindset

Required

4+ years of experience in machine learning engineering, ML infrastructure, or a related field
Demonstrate proficiency in programming languages such as Python, Java, C++, or Rust
Have 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, proactively identifying and solving complex infrastructure challenges
Collaborate effectively with cross-functional teams to deliver impactful solutions
Apply strong problem-solving skills and a growth mindset to continuously improve systems and processes

Benefits

We take care of you and create the conditions for you to do the best work of your career.
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