Heidi Health Logo

Heidi Health

Mid Level AI Models Engineer

Reposted 3 Hours Ago
Be an Early Applicant
Hybrid
Melbourne, Victoria, AUS
Mid level
Hybrid
Melbourne, Victoria, AUS
Mid level
As a Mid Level AI Models Engineer, you will fine-tune large-scale language models, run experiments, and implement advanced machine learning techniques to improve AI performance and efficiency.
The summary above was generated by AI
Who We Are

Healthcare needs a better rhythm: one that keeps care continuous and deeply human. Heidi is building an AI Care Partner that works alongside clinicians to make that possible.

We’re a team of doctors, engineers, designers, researchers, and creatives building tools that help clinicians stay focused on what matters most: their patients.

In just 18 months, Heidi has given back more than 18 million hours to healthcare professionals — supporting 73 million patient visits in 116 countries. Today, more than two million patient visits each week are powered by Heidi worldwide.

Backed by nearly $100 million in funding, we’re growing in the US, UK, Canada, and Europe, partnering with leading health systems including the NHS, Beth Israel Lahey Health, and Monash Health.

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.

  • Partner with product, medical knowledge and design teams to automate prompt engineering, build agentic AI, create model evaluation frameworks.

  • Build cutting-edge LLM features within Heidi using the latest tools and frameworks

  • 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 can optimise AI features for accuracy, latency, scalability, and cost with AI agent design & fine-tuning.

  • You’ve shipped AI and LLM-powered features to real users in a production environment.

  • 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.

The way we work

1. Build to Last

We design for safety and reliability so clinicians, patients, and our teams can trust what we build every day.

2. Own Your Practice

Ideas rise on merit, not title, and everyone shares responsibility for the standards we set together.

3. Move Fast, Stay Steady

We move quickly but never at the cost of trust. Progress only matters if people can depend on what we make.

4. Make Others Better

Honest feedback, steady support, and shared growth keep our teams improving together.

Why you will flourish with us

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

  • 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!

Heidi is dedicated to creating an equitable, inclusive, and supportive work environment that brings people together from diverse backgrounds, experiences, and perspectives. Our strength is in our differences.

We're proud to be an equal opportunity employer and welcome all applicants as we're committed to promoting a culture of opportunity for all.

Top Skills

Python
PyTorch
TensorFlow

Similar Jobs

3 Hours Ago
Remote or Hybrid
4 Locations
Senior level
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
The role involves deploying and supporting CrowdStrike's AI capabilities for threat detection and response, along with mentoring and customer engagement.
Top Skills: AIAWSAzureChefCitrixFalconGCPGroup PolicyIbm BigfixLogscaleMachine LearningMcafee EpoMssqlPuppetPythonRSccmSIEMSplunk
3 Hours Ago
Remote or Hybrid
5 Locations
Senior level
Senior level
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
The Platform Professional Services Manager will lead a team providing deployment, operational support, and customer engagement for CrowdStrike technologies, managing technical implementations and collaborating across teams to ensure successful delivery outcomes.
Top Skills: AWSAzureBashBatchChefCitrixGCPGroup PolicyIbm BigfixMcafee EpoPdqPowershellPsexecPuppetPythonSccm
3 Hours Ago
Easy Apply
Remote or Hybrid
Australia
Easy Apply
Senior level
Senior level
Artificial Intelligence • Cloud • Software
Serve as technical pre- and post-sales advisor for enterprise customers: design architectures for web and AI/Agentic workflows on Vercel, run evaluations and POCs, guide onboarding and migrations, optimize performance and cost, respond to security/compliance reviews, and produce reusable playbooks and developer-facing content.
Top Skills: Ai SdkAWSAzureEdge PlatformsGCPHeadless CmsJavaScriptLighthouseNext.JsTypescriptVercel

What you need to know about the Melbourne Tech Scene

Home to 650 biotech companies, 10 major research institutes and nine universities, Melbourne is among one of the top cities for biotech. In fact, some of the greatest medical advancements were conceptualized and developed here, including Symex Lab's "lab-on-a-chip" solution that monitors hormones to predict ovulation for conception, and Denteric's vaccine for periodontal gum disease. Yet, the thousands of people working in the city's healthtech sector are just getting started, to say nothing of the tech advancements across all other sectors.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account