Senior Machine Learning Engineer at Headspace

Position Overview

Machine Learning 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 Machine Learning. These applications include building conversational AI systems, healthcare assistance tools, and recommendation and personalization systems. In this team, you’ll be tasked with owning and delivering cutting-edge language-based ML applications that will power the core features of Headspace. You’ll have the opportunity to lead the vision, alignment, development, deployment, and evangelization of these solutions, helping to bring Headspace to the forefront of AI and to realize its mission to improve health and happiness of the world.

How your skills and passion will come to life at Headspace:

• Lead the development of complex, scalable AI models and applications from inception to production
• Drive impactful ML technology initiatives that will shape the delivery of and access to mental healthcare
• Provide technical leadership and be a role model for those pursuing technical career paths in AI/ML engineering
• Contribute to an engaged, inclusive technology community that values diversity and equity
• Promote the development of high-quality code through engagement in code review and constructive feedback
• Increase internal and external company visibility in the AI community through open source, talks/presentations, etc.

What you’ve accomplished:

• MS 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
• 3+ years of experience with modern NLP tools and machine learning libraries (scikit-learn, PyTorch, TensorFlow, spaCy)
• 3+ 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.
• 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

Nice to have:

• Professional experience with clinical and/or healthcare applications of machine learning
• Familiarity with current ML literature including optimization methods and agent-based models
• Fluency in OO design and programming in Python
• Experience with implementation of robust and highly scalable services
• Experience with AWS, including SageMaker, Lambda, S3, DynamoDB, IAM

Location

We are currently hiring this role remotely in the US and Hybrid for San Francisco (SF) and Los Angeles (LA). Candidates must permanently reside in the US full-time.

For candidates with a primary residence in the greater SF and LA areas, this role will follow our hybrid model. You’ll work 3 days per week from our office, allowing for impactful in-office collaboration and connection, while enjoying the flexibility of remote work for the rest of the week. Your recruiter will share more details about our hybrid model.

Pay & Benefits

The anticipated new hire base salary range for this full-time position is $122,400-$195,500 + equity + benefits.

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.

Application Instructions

If you’re excited by the idea of seeing yourself in this role at Headspace, please apply with your resume.