Staff Software Engineer, Machine Learning jobs in United States
cer-icon
Apply on Employer Site
company-logo

Headspace · 5 days ago

Staff Software Engineer, Machine Learning

Headspace is a company focused on transforming mental healthcare through technology. The Staff Software Engineer, Machine Learning will lead the development of the ML platform, driving key architectural decisions and ensuring the scalability and reliability of ML systems.

Artificial Intelligence (AI)FitnessHealth CareMental HealthmHealthMobile AppsPsychologyTherapeutics
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

Machine LearningAWS ServicesMLOpsPythonInfrastructure as CodeDevOpsCI/CD WorkflowsCode QualityProblem-SolvingTechnical LeadershipCommunication SkillsContinuous Learning

Required

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

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

Benefits

Equity
Comprehensive healthcare coverage
Monthly wellness stipend
Retirement savings match
Lifetime Headspace membership
Generous parental leave

Company

Headspace

company-logo
Headspace is a mental health company offering AI companionship, coaching, therapy, psychiatry, and mindfulness tools for everyday support.

H1B Sponsorship

Headspace has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3)
2024 (4)
2023 (3)
2022 (8)
2021 (4)
2020 (5)

Funding

Current Stage
Late Stage
Total Funding
$320.85M
Key Investors
Oxford Finance LLCBliscePacific Western Bank
2023-07-26Debt Financing· $105M
2022-02-14Series C
2021-10-31Secondary Market

Leadership Team

leader-logo
Tom Pickett
Chief Executive Officer
linkedin
leader-logo
Andy Puddicombe
Co-Founder
Company data provided by crunchbase