Maincode Logo

Maincode

AI Engineer

Reposted Yesterday
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

10 Days Ago
In-Office
Melbourne, Victoria, AUS
Expert/Leader
Expert/Leader
Artificial Intelligence • Information Technology • Software • Automation
The Lead AI Engineer will architect AI solutions, mentor teams, and engage with clients to develop and implement AI strategies, focusing on Generative AI.
Top Skills: LangchainPythonPyTorchSQLTensorFlow
6 Days Ago
In-Office
South Melbourne, Melbourne, Victoria, AUS
Mid level
Mid level
eCommerce • Retail
As an AI Engineer at Kogan.com, you'll design and deploy AI solutions, collaborate with teams, and improve customer experiences using Python and React.
Top Skills: AIMlPythonReact
17 Days Ago
In-Office
Melbourne, Victoria, AUS
Senior level
Senior level
Robotics • Software • Automation
The Simulation and Test Engineer will develop testing infrastructure for AI systems, focusing on conversational AI, perception, and gesture behavior while ensuring quality through automated regression tests and CI/CD pipelines.
Top Skills: C++Ci/Cd ToolsGithub ActionsJenkinsLlm TestingNvidia Isaac SimPytestPythonRos2Simulation PlatformsTesting FrameworksUnityUnreal EngineVoice Ai Systems

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