Hybrid Senior AI Models Engineer

at Heidi Health

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

About the role

• Senior AI Models Engineer at Heidi leading AI research and development initiatives. Focusing on improving model performance in healthcare domain with hybrid work environment.

Responsibilities

• Lead Model R&D Initiatives: Lead major research and development projects focused on step-changes in model performance, from ideation and experiment design to productionisation.
• Solve the Hardest Problems: Tackle the most ambiguous and challenging technical problems related to model behaviour, alignment, and safety.
• Build cutting-edge fine-tuned LLMs within Heidi using the latest tools and frameworks
• Partner with product, medical knowledge and design teams to automate prompt engineering, build agentic AI, create model evaluation frameworks.
• Develop Novel Techniques: Go beyond existing literature to develop novel fine-tuning or data generation techniques tailored to our unique healthcare domain.
• Mentor and Guide: Mentor mid-level and junior engineers, sharing your deep expertise and fostering a culture of scientific rigor and innovation.

Requirements

• You’ve a proven track record of leading research or development projects that have resulted in significant improvements to large-scale AI models.
• You’ve previous hands-on experience building scalable AI products or features.
• You’ve shipped AI and LLM-powered features to real users in a production environment.
• You are considered an expert in Python, deep learning frameworks (PyTorch/TensorFlow), and the underlying theory of transformer architectures.
• You have a world-class intuition for AI/ML, built on a deep and solid foundation in mathematics and statistics.
• You have the ability to translate vague product needs into concrete research questions and well-designed experiments.
• A PhD in Computer Science, AI, Statistics, or a related field is highly preferred.
• Bonus: A strong publication record in top-tier AI conferences.
• Bonus: Experience building and leading a research-focused team.
• Bonus: Expertise in reinforcement learning or data-centric AI.

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!