Software Engineer, ML Infra (Junior & New Grad)

Machine Learning Infrastructure Engineer

We're hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the models behind our Ads and Recommendations products. You'll join a small, high-ownership team that ships platform improvements end-to-end—partnering with product and data teams, reducing latency and cost, and shortening the path from an idea to a safely launched model.

You'll work across the ML lifecycle: making training faster and more reliable, improving model serving performance, and strengthening our feature/embedding platform so models stay fresh and consistent between offline and online use. We're looking for someone who can take real ownership, finish what's started, and raise the bar on stability and developer experience.

If you like building reliable systems that make ML teams move faster—and you enjoy turning complexity into simple, durable solutions—we'd love to talk.

Responsibilities

Design and develop machine learning infrastructure.

Own and enhance core components of the ML infrastructure, including systems for offline and online model training, model pipeline health monitoring, model serving, feature authoring, and feature serving.

Proactively address ML infrastructure issues that may impact production.

Collaborate with ML engineers to build robust model pipelines utilizing the ML infrastructure.

Requirements

Bachelor's degree in Computer Science, Engineering, or a related field with 2+ years of relevant work experience; or a Master's/PhD in a related discipline.

Familiarity with software development processes, including version control, bug tracking, and design documentation.

Proficient in Python, with a strong understanding of object-oriented languages such as C++ or Java

Basic knowledge of Applied Machine Learning and experience with major Deep Learning frameworks, such as PyTorch and TensorFlow.

Preferred qualifications

Familiarity with cloud services such as AWS, GCP, Azure, and others.

Contributions to open-source machine learning or infrastructure tools.

A systematic, data-driven problem-solving approach combined with strong communication skills.

The US base salary range for this full-time position is listed below. Pay may vary based on a number of factors including job-related skills, level, experience, geographic location and relevant education or training. At NewsBreak, we design our overall rewards package to attract top talents. Depending on the position, the role may also be eligible for discretionary bonus and options. Your recruiter can share more details during the hiring process.

Annual Base Pay Range

$125,000 - $175,000 USD