Staff Engineer, Machine Learning

About Us

One Medical is a primary care platform challenging the industry status quo by making quality care more affordable, accessible and enjoyable. But this isn't your average doctor's office. We're on a mission to radically transform healthcare, which means tackling the frustrations of everyone involved - from patients and providers to employers and health networks.

Across the country, our members enjoy seamless access to comprehensive care at more than 180 locations across 28 cities (and counting!) as well as 24/7 access to virtual care powered by intelligent uses of technology. In addition to a direct-to-consumer membership model, we work with more than 7,000 companies to provide One Medical health benefits to their employees.

On January 31, 2020 we marked a milestone with our public listing on Nasdaq, but our work is far from over. As we continue to grow and broaden our impact, we're building a diverse, driven and empathetic team, while working hard to cultivate an environment where everyone can thrive.

The Opportunity

At One Medical we build software to improve the healthcare experience for not just our members, but for our clinical staff as well. This enables us to build amazing end-to-end solutions to connect patients and our care team in new and innovative ways.

As a Staff Software Engineer (Machine Learning), not only will you be partnering with other ML engineers, but also designers and product managers. This role plays a significant role in designing and building our ML platform.

This role is particularly well-suited for someone who values having significant ownership of their work, helping to lead the direction of the team and/or partnering and mentoring other engineers. We tend to partner with other product teams to help them build solutions in their domains.

We are a collaborative and intellectually curious group. We have strong opinions yet keep them loosely held--it's important to us we create an environment where everyone feels safe and supported to voice their thoughts and ideas. If you buy into these ideologies, we'd love to find a time to connect.

What you'll likely work on:

  • ML Platform: Our vision is to enable ML usage across different domains and teams. Thus we are building an in-house ML platform which will enable us to understand and support our patients with ML products.
  • ML Infrastructure: Design and develop infrastructure for the full cycle of machine learning such as workflow management, feature store, data discovery tools, and feature libraries.
  • Scalability : The team is continuously improving and scaling our existing services.To help us achieve even improvements, we are building a platform which not only our own team but other teams will leverage to scale their data powered systems. The team is continuously improving and scaling our existing services. To help us achieve even more improvements, we are building a platform which not only our own team but also other teams will leverage to scale their data powered systems. Help us define the best practices and create ML systems to reach them.
  • Sustainability: The team is invested in continual innovation on the ML infrastructure itself to carefully orchestrate a continuous cycle of learning, inference, and observation while also maintaining high system availability and reliability.

Must have:

  • Solid engineering and infrastructure skills with 7+ years experience. We focus on a wide variety of use cases including NLP, demand forecasting, and churn predictions
  • Strong experience with Kubernetes and Docker
  • Experienced developing infrastructure on cloud providers such as AWS and GCP
  • Experience with ML pipelines architecture and design
  • Experience with services architecture and deploying ML services to production
  • Knowledge of security best practices
  • Proficient in Python
  • Excellent communication skills to collaborate with partners across the organization
  • Understanding of relational (MySQL, Postgres) and NoSQL databases (DynamoDB)
  • B.S. / M.S. in Computer Science, Electrical Engineering, or related experience

Nice to have:

  • Experience deploying ML applications such as Kubeflow, KServe, and MLflow
  • Experience deploying ML services built on Tensorflow, PyTorch or Transformers
  • Experience with Terraform or similar infrastructure-as-code tools
  • Experience with human-in-the-loop systems

Benefits designed to aid your health and wellness:

Taking care of you today
  • Paid sabbatical after 5 and 10 years
  • Employee Assistance Program - Free confidential advice for team members who need help with stress, anxiety, financial planning, and legal issues
  • Competitive Medical, Dental and Vision plans
  • Free One Medical memberships for yourself, your friends and family
  • Pre-Tax commuter benefits
  • PTO cash outs - Option to cash out up to 40 accrued hours per year

Protecting your future for you and your family
  • 401K match
  • Opportunity to participate in company equity programs
  • Credit towards emergency childcare
  • Company paid maternity and paternity leave
  • Paid Life Insurance - One Medical pays 100% of the cost of Basic Life Insurance
  • Disability insurance - One Medical pays 100% of the cost of Short Term and Long Term Disability Insurance

This is a full-time based role. The team is mostly based in the San Francisco Bay Area, but can be located anywhere in the United States.

One Medical

is committed to fair and equitable compensation practices. The base salary range for this role is $142,500 to $260,000 . Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to skill set, depth of experience, certifications, and specific work location. The total compensation package for this position may also include annual performance bonus, RSUs, benefits and/or other applicable incentive compensation plans. For more information, visit



One Medical is an equal opportunity employer, and we encourage qualified applicants of every background, ability, and life experience to contact us about appropriate employment opportunities.

Subject to applicable law, proof of COVID 19 vaccination is required for employees and contractors who interact with patients, access a shared office space or engage with other team members, except where a medical or religious accommodation applies.

One Medical participates in E-Verify and will provide the federal government with your Form I-9 information to confirm that you are authorized to work in the U.S. Please refer to the E-Verification Poster (English/Spanish) and Right to Work Poster (English/Spanish) for additional information.