Maincode Logo

Maincode

AI Engineer

Reposted 13 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 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 optimise inference systems for real-world deployment

  • Improve data pipelines, tokenisation, 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

13 Days Ago
In-Office
Melbourne, Victoria, AUS
Mid level
Mid level
Artificial Intelligence • Cloud • Information Technology • Consulting • Cybersecurity • Big Data Analytics
As a Generative AI Engineer, you will develop bespoke AI solutions, implement RAG patterns, manage CI/CD workflows, and engage with clients.
Top Skills: AWSAzureCi/CdCrewaiGCPGenerative AiGoogle Ai Development KitLangchainLanggraphPython
11 Hours Ago
In-Office
3 Locations
Junior
Junior
Logistics • Transportation
The Associate AI Solutions Engineer - UX will work on the machine learning lifecycle, develop AI products, and enhance user experiences through collaborative efforts and prototyping.
Top Skills: AWSFlutterGCPGoogle BigqueryPythonReactSnowflakeSQLVertex Ai
Yesterday
In-Office
Melbourne, Victoria, AUS
Senior level
Senior level
Logistics • Transportation
The Principal AI Platform Engineer will architect and maintain AI systems and infrastructure, drive technical roadmaps, and implement DevSecOps practices to enhance data security and performance.
Top Skills: AIAWSCi/CdDockerGCPKubernetesMachine LearningTerraform

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