Senior Machine Learning Engineer
About The Position
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.
Requirements
- 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 Haves
- 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.
Responsibilities
- 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.
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
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
Education Level
No Education Listed
