[Remote] Staff Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. Headspace is a company dedicated to providing mental health support through innovative technology. They are seeking a Staff Machine Learning Engineer to lead the development of AI systems that personalize user experiences, including content recommendations and conversational AI. This role involves technical leadership, collaborative problem-solving, and shaping the architecture of personalization systems.

Responsibilities

• Lead the development of recommender systems for Headspace meditation & mindfulness content as well as other backend services that enable a personalized member experience including search, a user knowledge graph, and conversational AI memory
• Drive impactful ML technology initiatives that will shape the delivery of and access to mental healthcare
• Serve as a go-to expert and mentor, exemplifying excellence in AI/ML engineering and inspiring others to pursue technical career growth
• Contribute to the design, development, and evolution of our AI systems, taking it from high-level vision to robust implementation, enabling production-ready AI capabilities
• Partner with our team of software engineers, ML engineers, and MLOps engineers, and our Product & Clinical leads to build high quality features that improve members’ lives

Skills

• Bachelor of Science degree or higher in Computer Science, Statistics, Mathematics or a related field OR equivalent experience
• 5+ years of ML engineering experience in an academic or professional setting, programming in Python
• 5+ years of experience with any of the following fundamental technologies: vector search, embedding models, recommender systems, supervised, unsupervised machine learning, deep learning, reinforcement learning, LLM orchestration, RAG systems
• 3+ years of experience with modern NLP tools and machine learning libraries (scikit-learn, PyTorch, TensorFlow, spaCy)
• Experience with unit, integration, and end-to-end testing, version control
• Strong problem solving and communication skills and ability to influence across internal organizations
• Mentorship of junior engineers and contribution to DEIB initiatives
• Master's degree in relevant field or equivalent experience
• Professional experience with clinical and/or healthcare applications of machine learning
• Familiarity with current ML literature
• Experience with implementation of robust and highly scalable services
• Experience with AWS, including SageMaker, Lambda, S3, DynamoDB, IAM

Benefits

• Equity
• Benefits
• Stock awards
• Comprehensive healthcare coverage
• Monthly wellness stipend
• Retirement savings match
• Lifetime Headspace membership
• Generous parental leave

Company Overview