Headspace aims to improve member and clinician experiences through the application of Machine Learning, specifically in areas like conversational AI, healthcare assistance, and personalization. The company is looking for a Staff Software Engineer to own and deliver cutting-edge language-based ML applications that will power core features and advance Headspace's mission in AI and mental healthcare.
Requirements
- 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
- Experience with implementation of robust and highly scalable services
- Experience with AWS, including SageMaker, Lambda, S3, DynamoDB, IAM
Responsibilities
- 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.
- Serve as a go-to expert and mentor, exemplifying excellence in AI/ML engineering and inspiring others to pursue technical career growth.
- Drive the design, development, and evolution of our internal ML platform, taking it from high-level vision to robust implementation.
- Partner with cross-functional teams to align technical decisions with organizational goals, ensuring cohesive and impactful solutions.
- Advocate for and contribute to high-quality engineering standards through rigorous code reviews and constructive, actionable feedback.
Other
- Candidates must permanently reside in the US full-time and be based in New York City and Seattle.
- 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
