Staff Machine Learning Engineer
About the role
The AI & Machine Learning group at Headspace is a dynamic and innovative group whose mission is to improve the experiences of our members and clinicians through the mindful application of AI. Our group builds conversational AI systems including Ebb, as well as search, recommendation & personalization systems that power the Headspace experience.
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.
- This will involve developing and deploying complex, scalable AI models and applications.
- 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.
- Shape Personalization Architecture: 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.
- Collaborative Problem-Solving: 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.
Requirements
- 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
Qualifications
- 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
Skills
- Experience with implementation of robust and highly scalable services
- Experience with AWS, including SageMaker, Lambda, S3, DynamoDB, IAM
Benefits
At Headspace, base salary is but one component of our Total Rewards package. We're proud of our robust package inclusive of: base salary, stock awards, comprehensive healthcare coverage, monthly wellness stipend, retirement savings match, lifetime Headspace membership, generous parental leave, and more. Additional details about our Total Rewards package will be provided during the recruitment process.
Pay & Benefits
The anticipated new hire base salary range for this full-time position is $140,400-195,000 + equity + benefits. For candidates based in San Francisco, New York City, or Seattle, a separate salary range of $179,400-$224,250 applies, consistent with our location-based compensation philosophy.
Our salary ranges are based on the job, level, and location, and reflect the lowest to highest geographic markets where we are hiring for this role within the United States. Within this range, individual compensation is determined by a candidate’s location as well as a range of factors including but not limited to: unique relevant experience, job-related skills, and education or training.
Your recruiter will provide more details on the specific salary range for your location during the hiring process.