Machine Learning Engineer - Large Language Models

This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for a Machine Learning Engineer - Large Language Models based in Australia.

Join a highly collaborative and innovation-driven environment where you'll help shape the future of AI-powered healthcare solutions. In this role, you'll design, optimize, and deploy advanced Large Language Models that support real-world applications across healthcare and life sciences. Working alongside multidisciplinary experts in machine learning, software engineering, and scientific domains, you'll contribute to cutting-edge AI initiatives with meaningful impact. This is an opportunity to solve complex technical challenges, leverage state-of-the-art technologies, and influence the development of production-ready AI systems. If you're passionate about LLMs, deep learning, and delivering scalable machine learning solutions, this role offers an exciting platform to grow your expertise while contributing to transformative projects.

Accountabilities: Fine-tune and optimize Large Language Models for a variety of healthcare and life science applications using techniques such as Supervised Fine-Tuning (SFT), Parameter-Efficient Fine-Tuning (PEFT), Direct Preference Optimization (DPO), and Proximal Policy Optimization (PPO).

Develop and enhance Retrieval-Augmented Generation (RAG) pipelines to improve the accuracy, relevance, and efficiency of AI-powered information retrieval systems.

Collect, prepare, clean, and curate high-quality datasets to support the training, evaluation, and continuous improvement of machine learning models.

Convert, optimize, and package AI models for production deployment across different serving environments while ensuring scalability and performance.

Collaborate with cross-functional teams to build, evaluate, and deploy robust deep learning solutions that address complex healthcare challenges.

Stay up to date with emerging advancements in LLMs, machine learning infrastructure, and AI engineering best practices to continuously improve model performance and reliability.

Requirements

At least 5 years of professional experience developing production-grade software engineering and deep learning solutions.

Degree in Computer Science, Data Science, Machine Learning, or a related technical field; a Master's or Ph.D. is strongly preferred.

Proven experience working with Large Language Models, including model fine-tuning and optimization techniques.

Hands-on expertise with frameworks such as Axolotl and strong knowledge of Transformer-based architectures.

Experience with machine learning frameworks, particularly PyTorch.

Familiarity with model serving technologies such as vLLM, Text Generation Inference (TGI), and llama.cpp.

Understanding of model quantization techniques and performance optimization for resource-constrained environments.

Strong problem-solving, collaboration, and communication skills with the ability to work effectively in distributed teams.

Passion for AI innovation and applying machine learning to impactful real-world use cases.

Benefits

Fully remote work environment with collaboration across an international team.

Competitive compensation package.

Opportunity to work on cutting-edge AI and Large Language Model technologies.

Continuous learning and professional development opportunities.

Exposure to impactful projects within healthcare, life sciences, and open-source AI.

Inclusive, diverse, and collaborative workplace committed to equal opportunity and fair hiring practices.

How Jobgether works: We use an AI-powered matching process to ensure your application is reviewed quickly, objectively, and fairly against the role's core requirements. Our system identifies the top-fitting candidates, and this shortlist is then shared directly with the hiring company. The final decision and next steps (interviews, assessments) are managed by their internal team. We appreciate your interest and wish you the best!  Why Apply Through Jobgether?    Data Privacy Notice: By submitting your application, you acknowledge that Jobgether will process your personal data to evaluate your candidacy and share relevant information with the hiring employer. This processing is based on legitimate interest and pre-contractual measures under applicable data protection laws (including GDPR). You may exercise your rights (access, rectification, erasure, objection) at any time.     #LI-CL1