Senior AI Models Engineer
Heidi Health
full-time
Posted on:
Location Type: Hybrid
Location: Melbourne • Australia
Visit company websiteExplore more
Job Level
Tech Stack
About the role
- 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!
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
Pythondeep learning frameworksPyTorchTensorFlowtransformer architecturesAI model performancefine-tuning techniquesdata generation techniquesmodel evaluation frameworksreinforcement learning
Soft Skills
mentoringguidingproblem-solvingscientific rigorinnovationcollaborationcommunicationtranslating product needsexperiment designleadership
Certifications
PhD in Computer SciencePhD in AIPhD in Statistics