Software Engineer, ML Infra (Junior & New Grad)

The role

You will design and develop machine learning infrastructure to train, serve, and monitor models for Ads and Recommendations products. You will work across the ML lifecycle to improve training speed, model serving performance, and feature platforms.

What you will do

• 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 servi
• Proactively address ML infrastructure issues that may impact production.
• Collaborate with ML engineers to build robust model pipelines utilizing the ML infrastructure.

Required qualifications

• 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.

Nice to have

• 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.

About the company

NewsBreak is a local life and news platform with more than 40 million monthly users. The company includes NewsBreak, Nearby AI, and NDot AI.