Staff Software Engineer / Tech Lead (ML Infrastructure)

Join HeartFlow, a pioneering company in precision healthcare. As a Staff Software Engineer / Tech Lead, you will lead a small team focused on Data and ML infrastructure, directly impacting large-scale machine learning applied to precision healthcare. This is a highly technical, hands-on role where you will set the architectural vision, guide day-to-day technical execution, and remain deeply involved in the codebase. You will work in a hybrid schedule out of our San Francisco office and enjoy a range of benefits including medical, dental, and vision coverage, a 401k plan with employer match, paid sick time and vacation, and more.

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