Heidi - Mid Level AI Models Engineer

• Own Model Fine-Tuning Cycles: Independently run end-to-end fine-tuning experiments, from data curation and setup to analysis and reporting. • Own Model Fine-Tuning Cycles: • Partner with product, medical knowledge and design teams to automate prompt engineering, build agentic AI, create model evaluation frameworks. • Partner with product, medical knowledge and design teams • Build cutting-edge LLM features within Heidi using the latest tools and frameworks • Build cutting-edge LLM features • Implement Advanced Techniques: Implement and adapt state-of-the-art algorithms from research papers to solve specific product challenges. • Implement Advanced Techniques: • Deep-Dive Model Analysis: Conduct in-depth analysis of model behaviour, identifying failure modes and proposing data- or algorithm-based solutions. • Deep-Dive Model Analysis: • Contribute to Core Infrastructure: Help build and improve the core training and experimentation infrastructure to increase research velocity. • Contribute to Core Infrastructure: • 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. • Python • PyTorch • 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 • The way we work • 1. Build to Last • 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. • 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.