Staff Machine Learning Engineer
hims & hers
AWSPythonPyTorch
Staff Machine Learning Engineer
Headway
full-time
Posted on:
Location Type: Remote
Location: Remote • California, New York, Washington • 🇺🇸 United States
Visit company websiteSalary
💰 $269,790 - $317,400 per year
Job Level
Lead
Tech Stack
PythonPyTorchScikit-LearnTensorflow
About the role
- 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
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
ATS Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
machine learningranking modelsrelevance modelssearch systemsrecommendation systemspersonalizationPythonA/B testingML frameworksmodel deployment
Soft skills
mentoringproduct intuitiontechnical depthfocus on patient experience
