Staff Software Engineer, Machine Learning
Posted on August 19, 2025 (about 1 month ago)
About the Staff Software Engineer, Machine Learning at Headspace:
The Headspace Machine Learning team plays a pivotal role in driving innovation, developing and launching key product features, and creating internal tools that power our mission. Our team is at the forefront of transforming mental healthcare through cutting-edge technology, leveraging the power of AI and ML to make a meaningful difference in people’s lives.
As a member of the ML Platform team, you will be at the heart of this transformation, building the foundational tools and systems that empower our ML engineers to deliver high-impact solutions efficiently and at scale. Your work will enable the team to develop, deploy, and maintain robust, resilient, and high-performing ML systems faster than ever before, creating a seamless pipeline from innovation to production.
In your role as a Staff Machine Learning Engineer, you will take the lead in shaping the vision and architecture of our core ML Platform. You will drive key decisions that define the future of how ML systems are built, scaled, and deployed at Headspace. By combining your technical expertise with strategic foresight, you will spearhead initiatives that enhance productivity, ensure scalability, and support our mission to improve the health and happiness of the world.
Location: This is a hybrid role located in our San Francisco office, working 3 days per week from the office.
What you will do:
- Shape ML Platform Architecture: Drive the design, development, and evolution of our internal ML platform, taking it from high-level vision to robust implementation.
- Engineer Scalable Systems: Build and support complex, scalable, and multi-component data and ML systems that integrate seamlessly across the organization.
- Automated Model Lifecycle Management: Develop frameworks for continuous retraining of production models, enabling online learning and adaptive system improvements.
- Collaborative Problem-Solving: Partner with cross-functional teams to align technical decisions with organizational goals, ensuring cohesive and impactful solutions.
- Technical Leadership: Serve as a go-to expert and mentor, exemplifying excellence in AI/ML engineering and inspiring others to pursue technical career growth.
- Champion Code Quality: Advocate for and contribute to high-quality engineering standards through rigorous code reviews and constructive, actionable feedback.
What you will bring:
Required Skills:
- Bachelor's or Master’s degree in Computer Science, Engineering, or a related field, or equivalent hands-on experience.
- Exceptional problem-solving abilities with the communication skills to influence decisions across teams and drive alignment.
- Proven track record in designing and delivering robust, scalable, and highly reliable services in production environments.
- A deep passion for technical excellence, continuous learning, and innovation.
- 5+ years of DevOps experience, with hands-on expertise in AWS services, including SageMaker, Lambda, S3, DynamoDB, and IAM.
- Proficiency with Infrastructure as Code (IaC) tools such as Terraform and development languages like TypeScript.
- Strong background in implementing unit, integration, and end-to-end testing, along with setting up and managing CI/CD workflows.
- 2+ years of experience in MLOps, including developing, standardizing, and automating machine learning workflows.
- Hands-on experience managing machine learning systems in production, ensuring reliability and scalability.
- Proficiency in object-oriented programming and design, particularly in Python.
- Experience driving ML initiatives from inception to deployment, with a focus on measurable business impact.
Preferred Skills:
- Strong knowledge of AI/ML technologies, including supervised and unsupervised learning, deep learning, and reinforcement learning.
- Hands-on experience building applications leveraging Large Language Models (LLMs).
- A genuine interest in applying technical skills to solve challenges in mental health, healthcare, or related fields.
Pay & Benefits:
The anticipated new hire base salary range for this full-time position is $140,400-$224,250 + 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. Individual compensation is determined by candidate’s location, relevant experience, skills, and education.
At Headspace, base salary is one component of a robust Total Rewards package including stock awards, healthcare coverage, wellness stipend, retirement savings match, lifetime membership, parental leave, and more.
About Headspace:
Headspace exists to provide every person access to lifelong mental health support through evidence-based content, clinical care, and innovative technology globally.
Values: Make the Mission Matter, Iterate to Great, Own the Outcome, and Connect with Courage. We work to build a connected, human-centered team redefining mental health care support.
Why You’ll Love Working Here:
- A mission that matters—with impact you can see and feel
- A culture that’s collaborative, inclusive, and grounded in our values
- The chance to shape what mental health care looks like next
- Competitive pay and benefits that support your whole self
Diversity, Equity, Inclusion and Belonging:
Headspace is committed to a diverse and inclusive work environment free of discrimination and harassment, fostering belonging and impactful work.
Equal opportunity employer, prohibits discrimination on protected grounds, respects EEOC laws, and supports accommodation for disabilities.
Applicants with disabilities may request accommodation through a specified form link.
Additional Information:
Headspace participates in the E-Verify Program.
Privacy and ethical guidelines related to dual relationships with mental health care providers are emphasized and managed carefully.
Apply for this job
You can apply by submitting your application through the Headspace Greenhouse job portal where this listing resides. Fill out all required fields marked with * and submit your resume/CV along with any other requested documents.
Additional assistance for applicants with disabilities is available by contacting the Talent team through the provided form link in the job description.