Research Machine Learning Engineer

About the Company

At Canva, we're building a future powered by AI that's as magical as it is impactful. Our CORE team within the Generative AI supergroup invents foundational technologies to power the future of AI-assisted design.

Responsibilities

• Design, implement, and optimize large-scale machine learning systems for training
• Improve GPU utilization, communication overhead, and memory efficiency
• Partner with research and modeling teams to align systems with algorithmic needs
• Evaluate and apply best practices for distributed training using industry-leading frameworks
• Perform low-level optimization including custom CUDA or Triton kernels
• Debug, profile, and fine-tune training workflows to unlock scalability

Requirements

• Strong background in LLMs, multimodal AI, or diffusion models
• Proficiency in Python
• Deep knowledge of PyTorch or JAX
• Experience with libraries such as Megatron-LM, NeMo, or DeepSpeed
• Familiarity with optimization techniques like FSDP/ZeRO, gradient checkpointing, or low-precision data types
• Hands-on experience writing custom GPU kernels in CUDA or Triton
• Full proficiency in English

Preferred Qualifications

• Familiarity with system programming languages such as C++ or Rust