Senior ML Engineer — Build Reliable Research Infra

The Onset; Sydney; On-site; AUD 120,000 - 180,000; Full time

Job summary

The Onset in Australia is seeking a senior ML engineer to translate experimental research into reliable software. You will implement and optimise ML models, build data and evaluation pipelines, and collaborate with researchers and engineers to scale training across cloud environments.

Ideal candidates have strong Python engineering skills, hands-on DL experience with PyTorch or JAX, and a track record turning research into usable software.

Qualifications

• Strong Python engineering skills and hands-on DL experience.
• Experience training and deploying deep learning models on cloud infrastructure.
• Proficiency with PyTorch or JAX in real projects.
• Ability to build dependable ML training, data or evaluation pipelines.
• Understanding of model performance, compute constraints and distributed workloads.
• Experience with large-scale model training including language or vision-language models.
• Familiarity with simulation environments, synthetic data or evaluation systems.
• Experience in a research lab, deep-tech company or ambitious AI startup.
• Evidence turning experimental work into usable software.
• Comfort working independently and adapting to evolving technology.

Responsibilities

• Turn experimental research code into reliable, efficient systems.
• Implement and optimise ML models and training pipelines.
• Build data, simulation and evaluation infrastructure for new research.
• Profile model behaviour and performance across compute environments.
• Improve reliability, scalability and repeatability of ML experiments.
• Build demonstrators that test research ideas in realistic settings.
• Support cloud infrastructure and services used by research and engineering teams.
• Work on distributed, real-time or embedded capabilities as the platform develops.
• Use AI-assisted development tools while applying strong engineering judgement.
• Help shape technical decisions in an evolving environment.

Skills

Python

Deep learning

PyTorch

JAX

Cloud training

Distributed training

ML pipelines

Research-to-prod

Independent work

Simulation data

Education

PhD helpful

Tools

PyTorch

JAX

Job description

The Onset in Australia is seeking a senior ML engineer to translate experimental research into reliable software. You will implement and optimise ML models, build data and evaluation pipelines, and collaborate with researchers and engineers to scale training across cloud environments.

Ideal candidates have strong Python engineering skills, hands-on DL experience with PyTorch or JAX, and a track record turning research into usable software.