Mid Level 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 an AI Model Engineer on the Model Platform team. You are a core contributor to advancing the capabilities of our foundational models. Your mission is to apply your deep technical expertise to run experiments, implement cutting-edge fine-tuning techniques (SFT, RLFT), and systematically improve the performance, safety, and efficiency of our LLMs.

What you'll do:

• Own Model Fine-Tuning Cycles: Independently run end-to-end fine-tuning experiments, from data curation and setup to analysis and reporting.

• Implement Advanced Techniques: Implement and adapt state-of-the-art algorithms from research papers to solve specific product challenges.

• Deep-Dive Model Analysis: Conduct in-depth analysis of model behaviour, identifying failure modes and proposing data- or algorithm-based solutions.

• Contribute to Core Infrastructure: Help build and improve the core training and experimentation infrastructure to increase research velocity.

What we will look for:

• You have hands-on experience training and fine-tuning large-scale language models in a professional or advanced academic setting.

• You have strong programming skills in Python and deep expertise with PyTorch or TensorFlow.

• You possess a strong intuition for machine learning and can debug and solve challenging problems in a complex model codebase.

• You have a strong background in statistics and a rigorous, data-driven approach to experimentation.

  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.