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

Headspace · 3 months ago

Staff Software Engineer, Machine Learning

Headspace is a company dedicated to providing lifelong mental health support through innovative technology. The Staff Software Engineer, Machine Learning will lead the development of AI models and applications, driving impactful technology initiatives to enhance mental healthcare delivery.

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

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

Qualification

Machine Learning EngineeringPythonNLP ToolsDeep LearningAWSProblem SolvingCommunication SkillsMentorship

Required

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

Preferred

Master's degree in relevant field or equivalent experience
Professional experience with clinical and/or healthcare applications of machine learning
Familiarity with current ML literature including optimization methods and agent-based models
Experience with implementation of robust and highly scalable services
Experience with AWS, including SageMaker, Lambda, S3, DynamoDB, IAM

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