Staff Machine Learning Engineer

Join to apply for the Staff Machine Learning Engineer - Dataset & Training Platform role at Canva.

Where and how you can work

Our flagship campus is in Sydney. We also have a campus in Melbourne and co-working spaces in Brisbane, Perth and Adelaide. But you have choice in where and how you work — we trust our Canvanauts to choose the balance that empowers them and their team to achieve their goals.

What you'd be doing in this role

As Canva scales, change continues to be part of our DNA. But we like to think that's all part of the fun. So this will give you the flavour of the type of things you'll be working on when you start, but this will likely evolve.

At the moment, this role is focused on:

• Architecting foundational AI Platform capabilities, making key technical decisions that impact how models are trained and deployed across Canva.
• Leading cross-team initiatives to consolidate and improve dataset and training capabilities, working with stakeholders across multiple engineering and research teams.
• Building and maintaining paved roads for high-performance distributed data processing and training systems, optimising for cost efficiency and developer experience.
• Driving technical strategy discussions, weighing platform stability needs against product velocity requirements.
• Mentoring other engineers and contribute to growing Canva's AI platform engineering capabilities.
• Solving complex technical challenges spanning multiple parts of the ML stack.

You're probably a match if

• You have over 5 years of experience building and scaling ML training systems, with hands-on experience in distributed training, model lifecycle management, and large-scale data processing.
• You have a proven track record of designing and implementing foundational AI/ML infrastructure that supports multiple teams and use cases.
• You have experience setting technical direction, making architectural decisions, and influencing engineering practices across organisations.
• You possess a strong understanding of distributed computing, container orchestration (Kubernetes), and cloud infrastructure (preferably AWS).
• You are fluent in Python with deep knowledge of ML frameworks (PyTorch, TensorFlow) and modern tools (W&B, Ray, Anyscale).
• You have a strong understanding of generative AI systems, including LLMs, multimodal models, and foundation model fine-tuning.
• You have the ability to balance long-term platform investments with immediate business needs, making pragmatic technical decisions.
• You have experience working with product teams, research, and other engineering specialties to understand and address diverse technical requirements.
• You have a track record of growing other engineers and contributing to technical culture and standards.
• You have experience with infrastructure-as-code and an understanding of performance optimisation.
• You consider GitOps principles for automation and deployment a plus.

About The Team

Canva's GenAI Platform Group is responsible for the delivery of Capabilities and Solutions which support ML and AI initiatives, from early ideation and prototyping, through to scaling to meet the needs of millions of Canva users in production. We empower thousands of engineers and product managers to deliver amazing product features which harness the power of cutting-edge technologies.

Dataset & Training Platform team specifically focuses on the foundational Capabilities that power model training, dataset management, and AI/ML workflows. We're building the infrastructure that enables Canva's AI-first future, supporting everything from generative design models to intelligent automation systems that serve millions of users worldwide.

What's in it for you?

Achieving our crazy big goals motivates us to work hard — and we do — but you'll experience lots of moments of magic, connectivity and fun woven throughout life at Canva, too. We also offer a range of benefits to set you up for every success in and outside of work.

Here's a Taste Of What's On Offer

• Equity packages — we want our success to be yours too
• Inclusive parental leave policy that supports all parents & carers
• An annual Vibe & Thrive allowance to support your wellbeing, social connection, office setup & more
• Flexible leave options that empower you to be a force for good, take time to recharge and support you personally

Check out lifeatcanva.com for more info.

Other Stuff To Know

We celebrate all types of skills and backgrounds at Canva — so even if you don't feel like your skills quite match what's listed above — we still want to hear from you

We see AI as a powerful amplifier of creativity and technology at Canva. We're evolving how we assess AI skills in our Technology hiring experience - you'll tackle interactive, real-time challenges that reflect the kind of work we do. In some interviews, you may also be asked to solve a problem using an AI tool to show how you approach challenges with tech by your side. Your recruitment partner will walk you through what to expect. We celebrate all types of skills and backgrounds at Canva so even if you don't feel like your skills quite match what's listed above - we still want to hear from you

Please note that interviews are conducted virtually.

Seniority level

• Mid-Senior level

Employment type

• Full-time

Job function

• Software Development

Industries

• Software Development

#J-18808-Ljbffr

---