Senior AI Model Engineer

Who are Heidi?

Heidi is on a mission to halve the time it takes to deliver world-class care.

We believe that by 2050, every clinician will practice with AI systems that free them from administrative burdens and increase the quality and accessibility of care to patients across the world.

Built for clinicians, by clinicians, at the core of Heidi is its people. We are an eclectic bunch of inventors, builders, scientists, nurses, doctors, mathematicians, designers, creatives, and high-agency executors.

We achieve in 6 months what it takes our competitors 4 years to do. In just 12 months, 20 million patient consults were supported by Heidi, and we’re now powering more than 1 million consults every week.

With our most recent $16.6MM round of funding from leading VC firms, we’re geared up to supercharge our ambitious global growth, starting with the US, Canada, UK and Europe - and we need great people like you to get there.

The Role

Working closely with our Engineering Manager, you’ll be a Senior AI Model Engineer on the Model Platform team. You are a technical leader responsible for solving our most critical model-related challenges.

Your mission is to drive significant breakthroughs in our model's capabilities, leading research and development efforts in areas like advanced fine-tuning (RLFT), alignment, and reasoning.

You will combine your deep ML intuition and research background to push the boundaries of what our models can do.

What you’ll do:

• Lead Model R&D Initiatives: Lead major research and development projects focused on step-changes in model performance, from ideation and experiment design to productionisation.

• Solve the Hardest Problems: Tackle the most ambiguous and challenging technical problems related to model behaviour, alignment, and safety.

• Develop Novel Techniques: Go beyond existing literature to develop novel fine-tuning or data generation techniques tailored to our unique healthcare domain.

• Mentor and Guide: Mentor mid-level and junior engineers, sharing your deep expertise and fostering a culture of scientific rigor and innovation.

What we will look for:

• You have a proven track record of leading research or development projects that have resulted in significant improvements to large-scale AI models.

• You are considered an expert in Python, deep learning frameworks (PyTorch/TensorFlow), and the underlying theory of transformer architectures.

• You have a world-class intuition for AI/ML, built on a deep and solid foundation in mathematics and statistics.

• You have the ability to translate vague product needs into concrete research questions and well-designed experiments.

• A PhD in Computer Science, AI, Statistics, or a related field is highly preferred.

  Bonus:

  • A strong publication record in top-tier AI conferences.

  • Experience building and leading a research-focused team.

  • Expertise in reinforcement learning or data-centric AI.

What do we believe in?

• We create unconventional solutions to difficult problems and we build them fast. We want you to set impossible goals and make them happen, think landing a rocket but the medical version.

• You'll be surrounded by a world-class team of engineers, medicos and designers to do your best work, inspired by our shared beliefs:

  • We will stop at nothing to improve patient care across the world.

  • We design user experiences for joy and ship them fast.

  • We make decisions in a flat hierarchy that prioritizes the truth over rank.

  • We provide the resources for people to succeed and give them the freedom to do it.

Why you will flourish with us 🚀?

• Flexible hybrid working environment, with 3 days in the office.

• Additional paid day off for your birthday and wellness days

• Special corporate rates at Anytime Fitness in Melbourne, Sydney tbc.

• A generous personal development budget of $500 per annum

• Learn from some of the best engineers and creatives, joining a diverse team

• Become an owner, with shares (equity) in the company, if Heidi wins, we all win

• The rare chance to create a global impact as you immerse yourself in one of Australia’s leading healthtech startups

• If you have an impact quickly, the opportunity to fast track your startup career!

Help us reimagine primary care and change the face of healthcare in Australia and then around the world.