Maincode Logo

Maincode

AI Engineer

Posted 5 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 sovereign AI models in Australia. We are training foundation models from scratch, designing new reasoning architectures, and deploying them on state-of-the-art GPU clusters. This is not fine-tuning someone else’s work. This is building from first principles.

As an AI/ML Engineer, you’ll be part of the team that makes this real. You’ll work on both sides of the problem: how models are trained and how they run in the world. You’ll design and build the systems that power large-scale training runs and efficient inference. You’ll work closely with our AI Researchers to implement the latest algorithms and ideas.

This is a deep engineering role. You’ll be writing a lot of code, instrumenting systems, optimizing performance, and debugging weird, messy edge cases in distributed training and model serving. If you love figuring out how foundation models really work under the hood, this is your team.

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

  • Design efficient inference systems that run these models in real-world environments

  • Optimize data pipelines, tokenization, batch prep, and distributed training at multi-node scale

  • Build deep observability into training and inference, improving performance, correctness, and efficiency

  • Debug and troubleshoot the hardest parts of model training and deployment, including distributed failures, GPU bugs, data inconsistencies, and scaling limits

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

  • Establish strong ModelOps practices to ensure reproducibility, reliability, and continuous improvement across the full lifecycle of foundation models, from initial experiments to production deployment

Who you are
  • Passionate about how models are built, trained, and run, especially large-scale foundation models

  • Driven by curiosity about model internals, training dynamics, and system-level performance

  • Excited to work across both training and inference, not just one side of the ML stack

  • Skilled in Python and ML frameworks like PyTorch or JAX. Familiarity with distributed compute (CUDA, Triton, NCCL, etc.) is a bonus but not required

  • Constantly learning, whether it’s reading open-source repos, replicating research ideas, or designing your own tools to explore a problem

  • Hands-on and determined. You like writing code, running experiments, and figuring things out

  • Motivated to help build sovereign AI capability here in Australia

Why Maincode

We are a small team building some of the most advanced AI systems in Australia. We are creating new foundation models from scratch, not just using what’s already out there.

We operate our own GPU clusters, run large-scale training, and work closely across research and engineering to push the frontier of what’s possible.

You'll be surrounded by people who:

  • Care about model internals, not just outputs

  • Build things that work, at scale

  • Take pride in learning, experimenting, and shipping

  • Want to help Australia build independent, 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
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
7 Hours Ago
In-Office
Melbourne, Victoria, AUS
Mid level
Mid level
Artificial Intelligence • Cloud • Information Technology • Consulting • Cybersecurity • Big Data Analytics
The Platform Engineer will enhance the GenAI platform, maintain cloud infrastructure, support AI/ML model deployment, and integrate ML workflows.
Top Skills: Ai PlatformAmazon SagemakerAWSBambooBigQueryCloud BuildCloud ComposerCloud FunctionsCloudFormationCloudtrailCloudwatchDataflowEksGCPGitGitlab RunnersGkeIamLambdaOpenapiPub/SubRest ApiS3Step FunctionsSwaggerTerraformVertex AiVpc
6 Days Ago
In-Office
Melbourne, Victoria, AUS
Entry level
Entry level
Edtech
As a Graduate AI Workflow Engineer, you'll build AI-powered workflows, creating MVP solutions, connecting APIs, and improving workflows for real business challenges.
Top Skills: AIAirtableAPIsAutomation LibrariesAWSAzureGCPHugging FaceN8NPower AutomatePythonPyTorchTensorFlowZapier

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