Staff Software Engineer, ML Developer

Summary

The job is for an experienced ML Engineer or someone who has worked closely with ML Engineers to improve the velocity and user experience of ML engineers at Stack AV. The role involves working with ML Engineers to understand their development workflow, automation of manual work, experiment management, collaboration with cross-functional teams, staying updated with the latest technologies and trends in ML, and promoting engineering excellence.

Requirements

• Education: Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field
• Experience: 5+ years of experience in ML Infra / MLE. Familiarity with infra / front end ideal
• Strong analytical and problem-solving skills
• Excellent verbal and written communication skills, with the ability to convey complex technical concepts to non-technical stakeholders

Responsibilities

• Work regularly with ML Engineers to understand their development workflow and improve it
• Automation of manual work is a large part of this job. Examples include converting manual work to automated pipelines (on Flyte) and creating metrics that can make automated go/no-go decisions on model deployment
• Experiment management using third-part tools (like MLFlow) and connecting this to the rest of development workflow (data infra and training infra)
• Work with open source teams to merge our changes upstream

Preferred Qualifications

(preferred) prior experience with offline ML pipelines and experiment management tooling. In particular, Flyte, MLFlow and Weights and Biases