Senior Machine Learning Engineer - Research Optimisation

Canva
Sydney
Workplace: HybridFull timeFunction: Data Science & Machine LearningSkills: ["Python","PyTorch","Kubernetes","Docker","CI/CD","GPU","Distributed training","Monorepo","CUDA","TorchScript","ONNX","Triton","TensorRT"]

Bridge between research and production to turn experiments into scalable, user-facing ML features. Produce deployable models in a monorepo, build libraries and services, and optimize PyTorch workloads across multi-GPU, multi-node setups while reducing training cost and latency. Enhance observability, CI/CD, and reproducible builds to accelerate research-to-user delivery at Canva.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Canva
Canva
2 months ago

Senior Machine Learning Engineer - Research Optimisation

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 6 hours agoStatus: Live

Job Summary

Bridge between research and production to turn experiments into scalable, user-facing ML features. Produce deployable models in a monorepo, build libraries and services, and optimize PyTorch workloads across multi-GPU, multi-node setups while reducing training cost and latency. Enhance observability, CI/CD, and reproducible builds to accelerate research-to-user delivery at Canva.
Location: Sydney
Workplace: Hybrid
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Productionise research models by refactoring, testing, containerising, and integrating into the monorepo for scalable reuse.
  • •Profile and optimise PyTorch training jobs, identifying bottlenecks across compute, memory, I/O, and networking.
  • •Improve distributed training setups (multi-GPU, multi-node) and help teams choose the right parallelism strategy.
  • •Build and maintain inference services, SDKs, and shared libraries with standardised pre/post-processing interfaces.
  • •Create multi-variant runners and rollout frameworks (feature flags, canaries, A/B testing, automated rollbacks).

Pay and Benefits

Equity and Bonus:Equity
Perks:EquityParental LeaveWellness StipendFlexible Leave

Key Requirements

  • •Strong Python software engineering fundamentals and experience turning notebooks/prototypes into production-grade services.
  • •Experience shipping ML systems in production (containers, APIs, CI/CD) within a monorepo environment.
  • •Hands-on experience optimizing PyTorch training or inference, profiling workloads, and reasoning about GPU memory, compute, and throughput.
  • •Comfort with containerised environments and Kubernetes to debug and improve ML workloads.
  • •Ability to refactor research code into clean abstractions with stable, well-documented interfaces.
Skills:PythonPyTorchKubernetesDockerCI/CDGPUDistributed trainingMonorepoCUDATorchScriptONNXTritonTensorRT
Languages:English
Tech Stack:PythonPyTorchKubernetesDockerCI/CDWekaLustreDeepSpeedFSDPDDPMegatronTorchScriptTritonTensorRTCUDA

Company Brief

Canva
Canva is an online visual communications platform that provides drag-and-drop design tools, templates, and media assets for creating presentations, social media graphics, videos, documents and more, serving individuals and large enterprises globally.
Industry: Enterprise Software
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Scaleup
Valuation: Decacorn (USD 10B+)
Funding: Series E+
Headquarters: Surry Hills, New South Wales, Australia
Founded: 2012
Glassdoor
Glassdoor: 3.9
WebsiteLinkedInGlassdoor