Hybrid Mid Level AI Models Engineer

at Heidi Health

• Job title: Mid Level AI Models Engineer
• Job type: Full Time
• Experience level: Mid level, Senior
• Salary: Not specified
• Degree requirement: Bachelor's Degree
• Tech skills: Python, PyTorch, Tensorflow
• Location requirements: Hybrid, Melbourne, Australia

About the role

• AI Model Engineer contributing to foundational model advancements at Heidi. Focus on fine-tuning large-scale language models in an innovative healthtech startup.

Responsibilities

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

Requirements

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

Benefits

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