Headway

Rewiring a new mental healthcare system for access and affordability.

Staff Machine Learning Engineer

Machine Learning EngineerMachine Learning EngineerFull TimeRemoteTeam 201-500Since 2019H1B SponsorCompany SiteLinkedIn

Location

California + 2 moreAll locations: California, New York, Washington

Posted

168 days ago

Salary

$269.8K - $317.4K / year

6 yrs expEnglishPythonPy TorchScikit LearnTensorflow

Job Description

• Lead development of Headway’s ranking and relevance models for patient-provider matching • Design and scale ML models for matching and ranking, search, recommendation, and personalization • Use patient signals, provider attributes, and outcomes data to improve matching accuracy • Establish ML infrastructure for experimentation, model training, deployment, and monitoring • Set technical direction, build the initial ML stack, and mentor junior ML engineers and data scientists • Partner with product, engineering, and data science to define success metrics, design A/B experiments, and influence product strategy

Job Requirements

  • 6+ years of experience in applied ML roles, with at least 1 year working on ranking, relevance, search, or recommender systems
  • Fluent in Python
  • Experience with ML frameworks (TensorFlow, PyTorch, Scikit-learn, CatBoost, etc.)
  • Taken ML models from research/prototype to production at scale
  • Comfortable designing and running A/B experiments and choosing offline and online metrics
  • Combine product intuition with technical depth and focus on patient experience
  • (Nice to have) Experience with matching systems or personalization
  • (Nice to have) Familiarity with modern retrieval techniques (vector search, embeddings, semantic search)
  • (Nice to have) Exposure to ML infrastructure such as feature stores, model monitoring, and retraining pipelines
  • (Nice to have) Experience with Metaflow, SageMaker, and Outerbounds

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

Related Job Pages

More Machine Learning Engineer Jobs

Machine Learning Engineer170 days ago
Full TimeRemoteTeam 1,001-5,000Since 2008H1B Sponsor

Principal ML Engineer building Workiva’s Generative AI platform and MLOps cloud infrastructure.

AWSCloudDockerJavaKubernetesPythonGo
United States
$204K - $326K / year

Senior Machine Learning Consultant

OneSix

Where data and AI innovation meets practicality.

Machine Learning Engineer170 days ago
Full TimeRemoteTeam 51-200Since 2013H1B No Sponsor

Senior ML consultant at OneSix building and deploying ML solutions for enterprise clients

AWSAzureCloudGoogle Cloud PlatformPythonPyTorchScikit-Learn
United States

Senior MLOps Engineer

Quilter

Design your next PCB design in hours, not weeks, with Quilter's generative AI for electronics.

Machine Learning Engineer175 days ago
Full TimeRemoteTeam 1-10H1B No Sponsor

Senior ML Ops Engineer building ML infrastructure for automated PCB design at Quilter.

AWSAzureCloudDockerGoogle Cloud PlatformKubernetesNode.jsTerraform
United States