Maincode Logo

Maincode

AI Engineer

Reposted 21 Days Ago
Be an Early Applicant
In-Office
Melbourne, Victoria
Mid level
In-Office
Melbourne, Victoria
Mid level
As an AI Engineer at Maincode, you will build and optimize training pipelines and inference systems for large-scale foundation models, collaborating closely with researchers on cutting-edge algorithms.
The summary above was generated by AI

Maincode is building an AI model factory. We’ve built Matilda, Australia’s first large language model trained entirely from scratch, and we’re developing new reasoning architectures beyond LLMs. We design and train models from first principles, deploy them at scale, and push the limits of what can be built here in Australia.


As an AI/ML Engineer, you’ll help make this possible. You’ll work on both sides of the stack: how models are trained, and how they run in the world. You’ll design and build systems for large-scale training and efficient inference, collaborating closely with AI Researchers to turn new ideas into production-grade systems.

This is a deep engineering role. You’ll write a lot of code, instrument distributed systems, optimise performance, and debug the complex, messy edge cases that come with large-scale model training and deployment.

What you’ll do
  • Build and scale training pipelines for large language models and other architectures

  • Design and optimize inference systems for real-world deployment

  • Improve data pipelines, tokenization, and distributed training across multi-node clusters

  • Build observability into training and inference to improve speed, efficiency, and correctness

  • Diagnose and resolve distributed training and GPU-related issues

  • Work closely with AI Researchers to translate cutting-edge algorithms into robust systems

  • Establish ModelOps practices for reliability, reproducibility, and continuous improvement

Who you are
  • Deeply interested in how models are built, trained, and run

  • Curious about model internals, training dynamics, and system performance

  • Comfortable working across both training and inference stacks

  • Skilled in Python and frameworks like PyTorch or JAX (familiarity with CUDA, Triton, or NCCL is a plus)

  • Hands-on, persistent, and motivated by solving hard problems

  • Constantly learning—reading papers, exploring repos, or experimenting with new tools

  • Excited to help shape the next generation of Australian AI capability

Why Maincode

We’re a small, focused team building advanced AI systems from the ground up. We run our own GPU clusters, train our own models, and collaborate closely across research and engineering to move fast and build real capability.

You’ll work with people who:
  • Care about how models actually work, not just what they output

  • Build things that run at scale

  • Value curiosity, rigour, and craftsmanship

  • Want to help Australia develop world-class AI systems

Top Skills

Cuda
Jax
Nccl
Python
PyTorch
Triton
HQ

Maincode Melbourne, Victoria, AUS Office

Melbourne, VIC, Australia, 3000

Similar Jobs

9 Days Ago
Easy Apply
In-Office
Melbourne, Victoria, AUS
Easy Apply
Senior level
Senior level
Fintech • Software
The Senior AI Engineer will design, implement, and maintain AI models, develop machine learning applications, and collaborate with data engineers to automate workflows and enhance company processes.
Top Skills: AdkAgents SdkAzureC#DbtLanggraphPythonPyTorchSklearnSnowflakeSpacyTensorFlow
13 Days Ago
Easy Apply
In-Office
Melbourne, Victoria, AUS
Easy Apply
Senior level
Senior level
Fintech • Software
Lead AI Engineer responsible for designing and implementing AI systems, developing machine learning applications, managing data pipelines, and collaborating with cross-functional teams.
Top Skills: AdkAgents SdkAzureDbtLanggraphPythonPyTorchSklearnSnowflakeSpacyTensorFlow
7 Days Ago
Easy Apply
In-Office or Remote
Melbourne, Victoria, AUS
Easy Apply
Senior level
Senior level
Software
The Senior Backend Engineer at Speechify will optimize product flows, ensure infrastructure scalability, and manage backend systems related to AI voice technology.
Top Skills: AWSAzureDockerGCPKubernetesPython

What you need to know about the Melbourne Tech Scene

Home to 650 biotech companies, 10 major research institutes and nine universities, Melbourne is among one of the top cities for biotech. In fact, some of the greatest medical advancements were conceptualized and developed here, including Symex Lab's "lab-on-a-chip" solution that monitors hormones to predict ovulation for conception, and Denteric's vaccine for periodontal gum disease. Yet, the thousands of people working in the city's healthtech sector are just getting started, to say nothing of the tech advancements across all other sectors.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account