Staff Software Engineer - ML Infrastructure

What You'll Do

• Build and maintain cloud infra and distributed systems for MLOps
• Build out internal training framework to make research with large distributed models easy and fun
• Design and develop systems to support Voxel's ML development, with a focus on computer vision applications.
• Provide technical guidance, mentorship, and project management support.
• Build distributed systems and pipelines for data management, ensuring scalability, reliability, and performance.
• Demonstrate a deep understanding of DevOps practices and apply them to ML operations.
• Utilize technologies such as Kubernetes and various CI/CD systems to optimize infrastructure deployment and management.
• Handle the complexities of designing and implementing ML infrastructure systems in a dynamic and fast-paced startup environment.
• Contribute to the overall impact of Voxel's AI-powered workplace safety solutions by delivering robust and efficient ML infrastructure.

Qualifications

• Must-Haves
• Bachelor's degree in Computer Science or a related field.
• Minimum 5+ years of experience in software engineering, with a focus on infrastructure design.
• Have experience designing large, highly available distributed systems with Kubernetes.
• Proven experience in designing complex systems and strong software engineering skills.
• Strong understanding and experience with distributed systems like Apache Spark, Ray etc. and infrastructure design.
• Proficiency in containerization technologies like Docker and orchestration platforms like Kubernetes.
• Demonstrated expertise in DevOps practices, with a focus on ML.
• Have worked on ML tools that researchers love


Nice-to-Haves:
• Knowledge of advanced ML operations techniques, such as model deployment and monitoring.
• Experience with pipeline automation and data management in ML workflows.Previous experience building ML systems and working on ML adjacent teams.