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
Headspace is a mental health company offering AI companionship, coaching, therapy, psychiatry, and mindfulness tools for everyday support. It was founded in 2010, and is headquartered in Santa Monica, California, USA, with a workforce of 501-1000 employees. Its website is https://www.headspace.com/.