Junior 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 and senior model engineers, you’ll be a Junior AI Model Engineer on the Model Platform team. This role is a deep dive into the core of our AI. Your mission is to support the team in experimenting with, fine-tuning, and analysing our language models. This is an incredible opportunity for a candidate with a strong theoretical background to gain hands-on experience in solving the hardest technical challenges at the heart of our product.

What you’ll do:

• Assist in Model Experiments: Support the design and execution of experiments for model fine-tuning (SFT, RLFT) and other adaptation techniques.

• Data Preparation & Analysis: Help prepare and curate high-quality datasets for training and evaluation, and assist in analysing experiment results.

• Implement Model Components: Write code for specific parts of the training and inference pipeline under the guidance of senior engineers.

• Contribute to Research: Stay up-to-date with the latest research papers and help the team explore novel techniques for improving model performance.

What we will look for:

• You have a very strong academic background in Computer Science, Statistics, Mathematics, or a related field.

• You possess a solid foundation in Python and a deep understanding of machine learning fundamentals, probability, and statistics.

• You have foundational knowledge of deep learning frameworks like PyTorch or TensorFlow.

• You have a powerful sense of curiosity, a passion for solving complex technical puzzles, and an eagerness to learn from a world-class team.

  Bonus:

• Internship or research experience in AI.

• Familiarity with transformer architectures.

• A completed or in-progress Master's degree or PhD.

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.