Research MLE (Training Optimization)大模型训练优化工程师

About the Group/Team

We're the CORE team within the Generative AI supergroup. Our mission is to invent foundational technologies that will power the future of AI-assisted design. From large-scale models to groundbreaking research, our team builds the technical core of Canva’s creative intelligence engine. We collaborate globally to ship research that makes a real impact—from smart editing to AI video tools—at massive scale.

About the Role/Specialty

As a Machine Learning Engineer, you’ll lead efforts to scale and optimize the training system for our large-scale multimodal and foundation models. You’ll design distributed training systems using Megatron-LM, NVIDIA NeMo, FSDP, and Triton—pushing the limits of performance across compute, memory, and communication layers. You'll sit at the intersection of systems and AI research, directly shaping how we train the models that will power Canva’s next generation of products.

What you’ll do (responsibilities)

You’ll design, implement, and optimize large-scale machine learning systems for training

You’ll improve all aspects of performance, including GPU utilization, communication overhead, and memory efficiency.

You’ll partner with research and modeling teams to align systems with algorithmic needs.