Staff Software Engineer / Tech Lead, ML Infrastructure at Heartflow | San Francisco | Rezi

Staff Software Engineer / Tech Lead, ML Infrastructure at Heartflow

Staff Software Engineer / Tech Lead, ML Infrastructure

Heartflow · San Francisco

1 months ago

Staff Software Engineer / Tech Lead, ML Infrastructure

Heartflow · San Francisco

2 months ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

We are seeking a Staff Software Engineer to serve as the technical anchor for a team focused on Data and ML infrastructure. This role directly supports large-scale machine learning for precision healthcare and is a hands-on, player-coach position involving architectural vision and technical execution.

Responsibilities

  • Act as the technical lead and mentor for a small, high-impact team of engineers, guiding system design, conducting code reviews, and unblocking technical hurdles.
  • Write high-performance Python code and utilize frameworks like Ray to architect and maintain large-scale distributed computing platforms for ML training and evaluation.
  • Spearhead the deployment of complex ML algorithms into highly available, scalable cloud environments, ensuring models run efficiently in production.
  • Design and integrate robust cloud-data systems to manage the lifecycle of massive, unstructured medical datasets.
  • Work cross-functionally with researchers and engineers to understand their model development processes and address their ML training, serving, and production monitoring needs.
  • Responsibly and securely utilize AI-powered development tools to accelerate engineering workflows.

Requirements

  • 8+ years of professional software engineering experience, with a strong focus on ML infrastructure, distributed systems, or MLOps.
  • A history of mentoring peers and leading technical projects.
  • Ability to write clear, well-tested, and scalable code.
  • Deep understanding of modern distributed computing architectures and optimization of heavy compute and cloud-data workloads (AWS, GCP, or Azure).
  • Familiarity with infrastructure as code (e.g., CDK, Terraform).
  • Familiarity with modern distributed computing frameworks and table formats like Ray, Kubernetes, and Apache Iceberg.
  • Knowledge of cross-language bindings for high-performance computing (e.g., C++/Python).
  • Prior experience in the healthcare domain, highly-regulated environments, or handling image-based algorithms is a huge plus.

Skills

  • Python
  • Ray
  • AWS
  • GCP
  • Azure
  • CDK
  • Terraform
  • Kubernetes
  • Apache Iceberg
  • C++

Location

  • San Francisco, CA

Work Type

  • Hybrid
  • Full-time

Experience Level

  • Staff
  • 8+ years

Salary/Compensations

  • $190,000 - $250,000 base, plus cash bonus and equity.

About the Company

  • Heartflow is a medical technology company advancing the diagnosis and management of coronary artery disease using cutting-edge technology.
  • The flagship product is an AI-driven, non-invasive cardiac test called the Heartflow FFRCT Analysis.
  • Heartflow is the first AI-driven non-invasive integrated heart care solution across the CCTA pathway.
  • Heartflow is a publicly traded company (HTFL) that has received international recognition for exceptional strides in healthcare innovation.
  • The product is cleared for use in the US, UK, Europe, Japan and Canada, and has been used for more than 500,000 patients worldwide.

Equal Opportunity

  • Heartflow is an Equal Opportunity Employer.
  • We are committed to a work environment that supports, inspires, and respects all individuals and do not discriminate against any employee or applicant because of race, color, religion, marital status, age, national origin, ancestry, physical or mental disability, medical condition, pregnancy, genetic information, gender, sexual orientation, gender identity or expression, veteran status, or any other status protected under federal, state, or local law.
  • This policy applies to every aspect of employment at Heartflow, including recruitment, hiring, training, relocation, promotion, and termination.