Salary
💰 $240,000 - $260,000 per year
Tech Stack
AWSPythonPyTorchTensorflow
About the role
- How can we use data to build systems that enable access to significantly better health care? In this role as Sr. Staff Machine Learning Engineer, you will define and lead the architecture and development of large-scale, foundational machine learning systems that power care personalization, treatment intelligence, and automation at scale. Your work will directly shape MedMatch—our AI-powered system—by owning the architecture and direction of key ML components, enhancing provider-patient interactions, optimizing treatment recommendations, and expanding into new verticals.
- Own and drive architectural decisions for ML and LLM models for treatment recommendation, personalization, and intelligent care delivery
- Work extensively with Python and ML libraries like PyTorch to build, optimize, and scale complex model pipelines
- Write and review high-quality, production-ready code supporting high-impact, cross-team ML systems
- Partner with engineers and product leadership to align ML development with strategic priorities
- Architect a robust ML infrastructure to enable streamlined training, evaluation, deployment, and monitoring
- Drive the adoption of new ML technologies and practices, contributing to long-term group-level technical maturity
- Mentor senior and staff-level engineers across multiple squads, guiding system-level thinking and best practices
- Champion a culture of scalable system design, engineering rigor, and customer-first innovation
Requirements
- 8+ years of experience in Machine Learning and/or Engineering, with a track record of technical leadership and ML system architecture
- Expert-level proficiency in Python and advanced ML frameworks such as PyTorch, TensorFlow
- Demonstrated success in architecting and maintaining ML platforms at scale across teams or business units
- Deep experience in LLMs and NLP, including transformer models, summarization, and chatbot applications
- Advanced understanding of ML infrastructure, including continuous training, real-time inference, and operational reliability
- Experience with ML infrastructure tools (e.g., Databricks, MLFlow, AWS SageMaker)
- A Master’s degree or PhD in Computer Science, Machine Learning, or a related field (not strictly required)
- Exceptional communication and cross-org leadership skills, with the ability to align diverse stakeholders around ML strategy