Senior Machine Learning Engineer

1 in 4 people in the US have a treatable mental health condition, but most providers don't accept insurance, making therapy too expensive for most people. Headway’s mission is to fix this by building a new mental healthcare system everyone can access. We started by solving the biggest barrier to care: insurance. The admin work - credentialing, claims, payment reconciliation - is a nightmare. We've automated that.

But we're going further. Over 70,000 providers across all 50 states run their practice on our software, serving over 1 million patients. We are building the best tools for therapists to run their entire practice, reimagining the experience of finding a therapist, and investing in the platform foundations to enable this at scale. We aren't just a billing layer; we are becoming the platform where care actually happens.

We're a Series D company with $325M+ in funding (a16z, Accel, GV, etc.), looking for exceptional people to help us achieve this mission. We want your time here to be the most meaningful experience of your career. Join us, and help change mental healthcare for the better.

About the team

The Ranking & Relevance team’s mission is to help every patient find the right provider for their needs. We are building the matching system that powers this connection, from search and discovery to ranking and personalization. Our goal is to combine cutting-edge machine learning with a deep understanding of patient and provider experience.

About the role

We are looking for a Senior Machine Learning Engineer to be a core contributor to Headway's ranking and relevance systems. You will design, build, and ship ML models that determine how patients discover and connect with therapists — using search, recommendation, and personalization techniques to improve matching quality across the care journey.

In this role, you will work with significant autonomy on scoped problem areas, contribute to technical direction, and collaborate closely with product, engineering, and data science to drive measurable outcomes for patients and providers.

What you’ll do

At Headway, one of our most important challenges is ensuring that patients find the right therapist quickly and effectively. We are investing deeply in search, ranking, and relevance systems powered by machine learning. As a Senior Machine Learning Engineer, you will focus on:

• Matching and ranking: Build and iterate on ML models that power how patients discover and connect with providers — from candidate generation to final ranking.
• Personalization: Leverage patient signals, provider attributes, and outcomes data to improve matching accuracy and relevance over time.
• Model development end-to-end: Own the full lifecycle of models you build — from offline evaluation and experimentation to production deployment and monitoring.
• Experimentation: Design and analyze A/B tests, define the right offline proxies and online metrics, and translate results into product decisions.
• Cross-functional collaboration: Partner with product managers, software engineers, and data scientists to scope problems, align on success metrics, and ship improvements that move the needle for patients.
• Raising the bar: Contribute to ML best practices on the team — through code reviews, documentation, and sharing what you learn from the systems you build.

Who you are

• You have 5+ years of experience in applied ML roles, with at least 3 years of meaningful hands-on modeling work in ranking, relevance, recommendations, search, or personalization.
• You are fluent in Python and experienced with ML frameworks such as TensorFlow, PyTorch, Scikit-learn, or CatBoost.
• You have taken ML models from prototype to production and understand what it takes to make them reliable, maintainable, and performant at scale.
• You are comfortable designing and running A/B experiments, selecting appropriate metrics, and interpreting results with rigor.
• You bring product intuition alongside technical depth — you care about making the patient experience meaningfully better, not just improving an offline metric.
• You work well in collaborative, cross-functional environments and can communicate technical tradeoffs clearly to non-ML stakeholders.

Nice to have

• Experience with search, discovery, matching systems or consumer facing 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.
• Background in healthcare, marketplace, or other B2C domains where user-provider or user-product fit is the core product problem.

Our interview process

After you apply to Headway, here are some details of what to expect during the interview process.

1. Initial screen: You’ll connect with someone in recruiting so you can learn more about the team, Headway’s mission and exciting growth, and we can get a better idea of your background.
2. First round: You’ll meet with an engineer on the team to do some live coding and learn more about the engineering team.
3. Final rounds: You’ll meet several more team members for technical and non-technical interviews and leave with a fuller picture of what it’s like to work at Headway.
4. References and the offer: Our favorite part of the process. We’ll send over all of the details, including specifics on employee equity, and congratulatory messages from excited future team members.

Compensation and Benefits:

The expected base pay range for this position is $218,500 - $273,125, based on a variety of factors including qualifications, experience, and geographic location. In addition to base salary, this role may be eligible for an equity grant, depending on the position and level.

We are committed to offering a comprehensive and competitive total rewards package, including robust health and wellness benefits, retirement savings, and meaningful ownership opportunities through equity. Compensation decisions are made holistically, ensuring fairness and alignment with market benchmarks while recognizing individual contributions and potential.

• Benefits offered include:
• 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

We believe a team's strength is in its people, and we cannot achieve this mission without a team that reflects the diversity of this problem – across race, ethnicity, gender, sexuality, age, national origin, religion, family status, disability, military status, and experience. Headway is committed to the full inclusion of all qualified individuals. As part of this commitment, Headway will ensure that persons with disabilities are provided with reasonable accommodations. If reasonable accommodation is needed to participate in the job application or interview process, to perform essential job functions, and/or receive other benefits and privileges of employment, please inform the recruiter when they contact you to schedule your interview.

Headway participates in E-Verify. To learn more, click here.

A notice to Headway applicants: To protect yourself against phishing and recruitment fraud, please note that Headway only accepts applications through our official careers page at https://headway.co/careers. Headway will never refer you to external websites, ask for payment or personal information, or conduct interviews via messaging apps. All official communication will come from a @findheadway.com email address. If you are contacted by someone claiming to be from Headway via an unofficial channel, please do not share any information and report it as spam.