Staff Software Engineer / Tech Lead, ML Infrastructure

Heartflow is a medical technology company advancing the diagnosis and management of coronary artery disease, the #1 cause of death worldwide, using cutting-edge technology. The flagship product—an AI-driven, non-invasive cardiac test supported by the ACC/AHA Chest Pain Guidelines called the Heartflow FFRCT Analysis—provides a color-coded, 3D model of a patient’s coronary arteries indicating the impact blockages have on blood flow to the heart. Heartflow is the first AI-driven non-invasive integrated heart care solution across the CCTA pathway that helps clinicians identify stenoses in the coronary arteries (RoadMapAnalysis), assess coronary blood flow (FFRCT Analysis), and characterize and quantify coronary atherosclerosis (Plaque Analysis). Our pipeline of products is growing and so is our team; join us in helping to revolutionize precision heartcare.Heartflow is a publicly traded company (HTFL) that has received international recognition for exceptional strides in healthcare innovation, is supported by medical societies around the world, cleared for use in the US, UK, Europe, Japan and Canada, and has been used for more than 500,000 patients worldwide. Location: Hybrid / San Francisco, CAWhat's the role?We are looking for a Staff Software Engineer to act as the technical anchor for a small, focused team working on Data and ML infrastructure. Your work will directly support large-scale machine learning applied to precision healthcare.This is a highly technical, player-coach role. You will set the architectural vision and guide the day-to-day technical execution of the team, remaining deeply hands-on in the codebase. Whether you are already a seasoned tech lead or a Staff Individual Contributor (IC) ready to step up and guide a sub-team of engineers, this position gives you the platform to lead technically.This is a full-time position operating on a hybrid schedule out of our San Francisco office.What's this team like at Heartflow?We build Data & ML infrastructure to simplify developing, evaluating, and deploying algorithms on massive medical imaging datasets. Our team owns critical platforms across the stack: data systems focused on curation and analytics, as well as our core ML environment for both training and inference.We design our infrastructure to not just be highly performant, but also easy to use. We are proponents of self-serve interfaces and robust user documentation. Our team tackles challenging problems using first-principles thinking, and we greatly value the ability to break down complex architectural challenges into digestible, well-communicated solutions.What does the day-to-day look like?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 how they develop models, using that understanding to solve their ML training, serving, and production monitoring needs.Responsibly and securely utilize AI-powered development tools (like coding assistants or LLMs) to accelerate the team's engineering workflows.Qualities that will help you thrive in this role are: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, and an excitement to act as the technical anchor for a small team of engineers.Ability to write clear, well-tested, and scalable code, with high proficiency in Python.Deep understanding of modern distributed computing architectures and how to optimize 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.Salary Range: $190,000 - $250,000 base, plus cash bonus and equity.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.Positions posted for Heartflow are not intended for or open to third party recruiters / agencies. Submission of any unsolicited resumes for these positions will be considered to be free referrals.Heartflow has become aware of a fraud where unknown entities are posing as Heartflow recruiters in an attempt to obtain personal information from individuals as part of our application or job offer process. Before providing any personal information to outside parties, please verify the following: A) all legitimate Heartflow recruiter email addresses end with “@heartflow.com” and B) the position described is found on our careers site at .