Talent.com
Staff Machine Learning Engineer
Staff Machine Learning EngineerHeadway • San Francisco, California, United States
Staff Machine Learning Engineer

Staff Machine Learning Engineer

Headway • San Francisco, California, United States
30+ days ago
Job type
  • Full-time
Job description

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.

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  Staff Machine Learning Engineer to lead the development of Headway’s ranking and relevance models. You will design and build the systems that decide how patients and providers are matched, using search, recommendation, and personalization to help patients through their care journey. You will set technical direction, build the initial stack, and mentor future ML teammates.

What you’ll do

At Headway, we have built the largest network of therapists who accept insurance in the country. One of our most important challenges is ensuring that patients find the right therapist for them quickly and effectively. To do this, we are investing deeply in search, ranking, and relevance systems powered by machine learning. As a Staff Machine Learning Engineer on the Patient team, you will focus on :

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

Who you are

  • You have 6+ years of experience in applied ML roles, with at least 1 year working on ranking, relevance, search, or recommender systems.
  • You are fluent in Python and have experience with ML frameworks (TensorFlow, PyTorch, Scikit-learn, CatBoost, etc.).
  • You have taken ML models from research / prototype to production at scale.
  • You are comfortable designing and running A / B experiments, and you know how to choose the right offline and online metrics.
  • You combine product intuition with technical depth. You care about making the patient experience meaningfully better, not just optimizing a model score.
  • Nice to have

  • 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.
  • Our interview process

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

  • 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.
  • First round : You’ll meet with an engineer on the team to do some live coding and learn more about the engineering team.
  • 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.
  • 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 & Benefits :

    The expected base pay range for this position is $269,790 – $317,400, based on a variety of factors including qualifications, experience, and geographic location. In addition to base salary, this role may be eligible for performance-based variable compensation and 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 employees work remotely across the US, with the option to work from offices in New York City, San Francisco and Seattle. Headway participates in E-Verify. To learn more, click  here.

    Create a job alert for this search

    Staff Machine Learning Engineer • San Francisco, California, United States

    Related jobs
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Adobe • San Jose, CA, United States
    Full-time
    Staff Machine Learning Engineer.Adobe Photoshop is seeking a Staff Machine Learning Services Engineer to serve as the technical lead for our Generative AI Services domain.In this high-impact role, ...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning Engineer, Level 6

    Staff Machine Learning Engineer, Level 6

    Snap Inc. • Palo Alto, CA, United States
    Full-time
    Staff Machine Learning Engineer, Level 6 page is loaded.Staff Machine Learning Engineer, Level 6.Apply locations Palo Alto, California Seattle, Washington Santa Monica - 2850 Ocean Park Blvd San Fr...Show more
    Last updated: 17 days ago • Promoted
    Staff Machine Learning Engineer, AI

    Staff Machine Learning Engineer, AI

    Sentry • San Francisco, CA, United States
    Full-time
    Bad software is everywhere, and we’re tired of it.Sentry is on a mission to help developers write better software faster so we can get back to enjoying technology. With more than $217 million in fun...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Hive • San Francisco, CA, United States
    Full-time
    Hive is the leading provider of cloud-based AI solutions to understand, search, and generate content, and is trusted by hundreds of the world's largest and most innovative organizations.The company...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Suki • Redwood City, CA, United States
    Full-time
    Get AI-powered advice on this job and more exclusive features.The Future of Healthcare Needs You.At Suki, were building technology that listens, understands, and gets out of the way so clinicians c...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Staff Machine Learning Engineer, Homefeed

    Sr. Staff Machine Learning Engineer, Homefeed

    Pinterest • San Francisco, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Moloco, Inc. • Redwood City, CA, United States
    Full-time
    Moloco is a machine learning company empowering organizations of all sizes to grow and unlock the full value of their unique first-party data, elevating the traditional path to performance advertis...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning R&D Engineer

    Staff Machine Learning R&D Engineer

    Visual Lease • Sunnyvale, CA, United States
    Full-time
    Matterport is leading the digital transformation of the built world.Our groundbreaking spatial computing platform turns buildings into data making every space more valuable and accessible.Millions ...Show more
    Last updated: 30+ days ago • Promoted
    AIML- Staff ML Engineer, Machine Learning Platform Technologies

    AIML- Staff ML Engineer, Machine Learning Platform Technologies

    Apple Inc. • San Francisco, CA, United States
    Full-time
    AIML- Staff ML Engineer, Machine Learning Platform Technologies.San Francisco Bay Area, California, United States Machine Learning and AI. Do you want to make Apple products more intelligent for our...Show more
    Last updated: 3 days ago • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    GEICO • Palo Alto, CA, United States
    Full-time
    Staff Machine Learning Engineer.As a Staff Machine Learning Engineer, you will be the overall tech lead of a single AI / Machine Learning team, responsible for the tech design and tech health of the ...Show more
    Last updated: 3 hours ago • Promoted • New!
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Lily AI • San Francisco, CA, United States
    Full-time
    Staff Machine Learning Engineer.Staff Machine Learning Engineer.Lily AI is a female-founded retail AI company empowering retailers and brands by bridging the gap between merchant-speak and customer...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning Engineer, Acceleration

    Staff Machine Learning Engineer, Acceleration

    Waymo • San Francisco, CA, United States
    Full-time
    Staff Machine Learning Engineer, Optimization.Staff Machine Learning Engineer, Optimization.Staff Machine Learning Engineer, Optimization. Be among the first 25 applicants.Staff Machine Learning Eng...Show more
    Last updated: 4 days ago • Promoted
    Staff Machine Learning R&D Engineer

    Staff Machine Learning R&D Engineer

    Matterport • Sunnyvale, CA, United States
    Full-time
    Matterport is leading the digital transformation of the built world.Our groundbreaking spatial computing platform turns buildings into data making every space more valuable and accessible.Millions ...Show more
    Last updated: 8 days ago • Promoted
    Staff Machine Learning Engineer, Intelligent Scheduling Systems

    Staff Machine Learning Engineer, Intelligent Scheduling Systems

    Tesla • Fremont, CA, United States
    Full-time
    Staff Machine Learning Engineer, Intelligent Scheduling Systems.Be among the first 25 applicants.Staff Machine Learning Engineer, Intelligent Scheduling Systems. Tesla is seeking a Machine Learning ...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Staff Machine Learning Engineer, Closeup Relevance

    Sr. Staff Machine Learning Engineer, Closeup Relevance

    Pinterest • Palo Alto, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show more
    Last updated: 4 days ago • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    Quantcast • San Francisco, CA, US
    Full-time
    At Quantcast, we're redefining what's possible in digital advertising.As a global Demand Side Platform (DSP) powered by AI, we help marketers connect with the right audiences and deliver measurable...Show more
    Last updated: 6 days ago • Promoted
    AIML - Staff Machine Learning Engineer

    AIML - Staff Machine Learning Engineer

    Apple Inc. • Cupertino, CA, United States
    Full-time
    Cupertino, California, United States Machine Learning and AI.The Apple Knowledge & Information (AKI) Entity Resolution team is looking for senior and staff engineers to lead software projects suffu...Show more
    Last updated: 30+ days ago • Promoted
    Staff ML Engineer : Build & Deploy Scalable ML Pipelines

    Staff ML Engineer : Build & Deploy Scalable ML Pipelines

    Intuit Inc. • Mountain View, CA, United States
    Full-time
    A leading company in financial software is seeking a Staff Machine Learning Engineer to join their innovative team.In this role, you'll collaborate with highly skilled AI scientists and engineers t...Show more
    Last updated: 2 days ago • Promoted