Staff Software Engineer, Machine Learning - Remote

Overview

The Headspace Machine Learning team plays a pivotal role in driving innovation, developing and launching key product features, and creating internal tools that power our mission. Our team is at the forefront of transforming mental healthcare through cutting-edge technology, leveraging the power of AI and ML to make a meaningful difference in people’s lives.

In Short

• Shape ML Platform Architecture: Drive the design, development, and evolution of our internal ML platform, taking it from high-level vision to robust implementation.
• Engineer Scalable Systems: Build and support complex, scalable, and multi-component data and ML systems that integrate seamlessly across the organization.
• Automated Model Lifecycle Management: Develop frameworks for continuous retraining of production models, enabling online learning and adaptive system improvements.
• Collaborative Problem-Solving: Partner with cross-functional teams to align technical decisions with organizational goals, ensuring cohesive and impactful solutions.
• Technical Leadership: Serve as a go-to expert and mentor, exemplifying excellence in AI/ML engineering and inspiring others to pursue technical career growth.

Requirements

• Bachelor's or Master’s degree in Computer Science, Engineering, or a related field, or equivalent hands-on experience.
• 5+ years of DevOps experience, with hands-on expertise in AWS services, including SageMaker, Lambda, S3, DynamoDB, and IAM.
• 2+ years of experience in MLOps, including developing, standardizing, and automating machine learning workflows.
• Proficiency in object-oriented programming and design, particularly in Python.
• Strong knowledge of AI/ML technologies, including supervised and unsupervised learning, deep learning, and reinforcement learning.

Benefits

• Comprehensive healthcare coverage.
• Monthly wellness stipend.
• Retirement savings match.
• Lifetime Headspace membership.
• Unlimited, free mental health coaching.