AI Model Engineer contributing to foundational model advancements at Heidi. Focus on fine - tuning large - scale language models in an innovative healthtech startup.
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!
Job title
Job type
Experience level
Salary
Not specified
Degree requirement
Tech skills
Location requirements
Supervisor in RSM’s Advanced Analytics and AI Consulting practice focusing on data, machine learning, and AI solutions. Collaborating with teams to deliver advanced analytics projects for clients.
AI Co - Op supporting AI tools and workflows development within HR at Teradyne. Collaborating with Talent Acquisition and other HR teams on various AI initiatives.
Software Dev AI/ML Co - op developing cloud solutions and AI/ML workflows at Nokia. Work on Container Technology and Kubernetes while collaborating with a global R&D team.
AI Risk Business Partner at Bankwest ensuring sustainable AI innovation without exposing the bank to unacceptable risks. Shaping risk strategies and governance for AI initiatives.
AI Workforce Consultant at KLA designing AI literacy experiences and enabling adoption across teams. Focused on responsible AI use and transformation of workforce capabilities.
Product Strategy Director at Collins Aerospace leading C2 and AI foundational enabler components for Connected Battlespace Strategy. Driving growth, innovation, and competitive advantage in aerospace and defense.
Senior Technology Specialist in AI working at Digital Catapult, applying machine learning techniques in operational systems across various sectors. Collaborating with experts to enhance AI deployment in the UK Digital Twin Centre programme.
AI Deployment Lead driving adoption of Avra platform in large Brazilian companies. Engaging with clients to identify operational challenges and provide strategic insights.
AI Technical Trainer responsible for delivering high - quality AI and Machine Learning curriculum. Mentoring learners and improving learning products within an ambitious EdTech company.