Staff Machine Learning Engineer

Headway • San Francisco, CA

92d • $269,790 - $317,400

About The Position

Headway’s mission is a big one – to build a new mental health care system everyone can access. We’ve built technology that helps people find great therapists with the first software-enabled national network of providers accepting insurance. 1 in 4 people in the US have a treatable mental health condition, but the majority of providers don’t accept insurance, making therapy too expensive for most people. Headway is building a new mental healthcare system that everyone can access by making it easy for therapists to accept insurance and scale their practice. Headway was founded in 2019. Since then, we’ve grown into a diverse, national network of over 60,000 mental healthcare providers across all 50 states who run their practice on our software and have served over 1 million patients. We’re a Series D company with over $325m in funding from a16z (Andreessen Horowitz), Accel, GV (formerly Google Ventures), Spark Capital, Thrive Capital, Forerunner Ventures and Health Care Service Corporation. We want your time here to be the most meaningful experience of your career. Join us, and help change mental healthcare for the better.

Requirements

• 6+ years of experience in applied ML roles, with at least 1 year working on ranking, relevance, search, or recommender systems.
• Fluency in Python and experience with ML frameworks (TensorFlow, PyTorch, Scikit-learn, CatBoost, etc.).
• Experience taking ML models from research/prototype to production at scale.
• Comfortable designing and running A/B experiments, and knowledgeable about choosing the right offline and online metrics.
• Combination of product intuition with technical depth, caring about making the patient experience meaningfully better.

Nice To Haves

• Experience with matching systems or personalization.
• Familiarity with modern retrieval techniques (vector search, embeddings, semantic search).
• Exposure to ML infrastructure such as feature stores, model monitoring, and retraining pipelines.
• Experience with Metaflow, SageMaker, and Outerbounds.

Responsibilities

• Design and scale ML models that power how patients discover and connect with providers.
• Use patient signals, provider attributes, and outcomes data to improve matching accuracy.
• Establish the foundation for experimentation, model training, deployment, and monitoring across Headway.
• Guide junior ML engineers and data scientists as the team grows, setting the bar for ML practices at Headway.
• Partner with product, engineering, and data science to define success metrics, design A/B experiments, and influence product strategy.

Benefits

• Equity compensation
• Medical, Dental, and Vision coverage
• HSA / FSA
• 401K
• Work-from-Home Stipend
• Therapy Reimbursement
• 16-week parental leave for eligible employees
• Carrot Fertility annual reimbursement and membership
• 13 paid holidays each year as well as a Holiday Break during the week between December 25th and December 31st
• Flexible PTO
• Employee Assistance Program (EAP)
• Training and professional development

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Number of Employees: 501-1,000 employees