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 Guides
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
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
QuilterDesign 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