Software Engineer, ML Infra

$165,000 - $165,000/Yr

Newsbreak - Mountain View, CA

Full-time - Mid Level

Mountain View, CA

Sporting Goods, Hobby, Musical Instrument, Book, and Miscellaneous Retailers

About the position

The Software Engineer, ML Infra position at NewsBreak focuses on designing and developing machine learning infrastructure to enhance the company's capabilities in local news delivery. The role involves owning and improving core components of the ML infrastructure, ensuring robust model pipelines, and collaborating with ML engineers to address infrastructure issues that may affect production.

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 programming and microservices architecture.
• Basic knowledge of Applied Machine Learning and experience with major Deep Learning frameworks, such as PyTorch and TensorFlow.

Nice-to-haves

• Proven track record in building large-scale distributed backend systems.
• 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.

Benefits

• Health, dental, and vision care for you and your family
• Top-tier 401(K) plan with company matching
• Paid time off and paid holidays
• Paid parental leave
• FSA and commuter benefits programs
• Team activity budget