Software Engineer, ML Infra

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.

Preferred qualifications

• 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