Track This Job
Add this job to your tracking list to:
- Monitor application status and updates
- Change status (Applied, Interview, Offer, etc.)
- Add personal notes and comments
- Set reminders for follow-ups
- Track your entire application journey
Save This Job
Add this job to your saved collection to:
- Access easily from your saved jobs dashboard
- Review job details later without searching again
- Compare with other saved opportunities
- Keep a collection of interesting positions
- Receive notifications about saved jobs before they expire
AI-Powered Job Summary
Get a concise overview of key job requirements, responsibilities, and qualifications in seconds.
Pro Tip: Use this feature to quickly decide if a job matches your skills before reading the full description.
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
- 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
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
Key Skills
Ranked by relevanceReady to apply?
Join Maincode and take your career to the next level!
Application takes less than 5 minutes