
Staff Software Engineer, Machine Learning
headspace
full-time
Posted on:
Location Type: Hybrid
Location: California • New York • United States
Visit company websiteExplore more
Salary
💰 $140,400 - $224,250 per year
Job Level
Tech Stack
About the role
- Lead the development of complex, scalable AI models and applications from inception to production.
- Drive impactful ML technology initiatives that will shape the delivery of and access to mental healthcare.
- Partner with cross-functional teams to align technical decisions with organizational goals, ensuring cohesive and impactful solutions.
- Advocate for and contribute to high-quality engineering standards through rigorous code reviews and constructive, actionable feedback.
Requirements
- Bachelor of Science degree or higher in Computer Science, Statistics, Mathematics or a related field OR equivalent experience
- 5+ years of ML engineering experience in an academic or professional setting, programming in Python
- 5+ years of experience with any of the following fundamental technologies: vector search, embedding models, recommender systems, supervised, unsupervised machine learning, deep learning, reinforcement learning, LLM orchestration, RAG systems.
- 3+ years of experience with modern NLP tools and machine learning libraries (scikit-learn, PyTorch, TensorFlow, spaCy)
- Experience with unit, integration, and end-to-end testing, version control
- Strong problem solving and communication skills and ability to influence across internal organizations
- Mentorship of junior engineers and contribution to DEIB initiatives
Benefits
- base salary
- stock awards
- comprehensive healthcare coverage
- monthly wellness stipend
- retirement savings match
- lifetime Headspace membership
- generous parental leave
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
machine learningdeep learningreinforcement learningnatural language processingPythonvector searchembedding modelsrecommender systemsscikit-learnPyTorch
Soft Skills
problem solvingcommunicationinfluencementorshipcollaborationfeedbackadvocacy