About this role
Machine Learning Engineer (Training Optimization）大模型训练优化工程师 at Canva. Location: Beijing, Beijing, China. Role: designing systems, optimizing training, debugging workflows Requirements: Experience scaling and optimizing distributed training for large multimodal models; strong ML systems skills, Python, PyTorch/JAX, CUDA or Triton kernel experience, and excellent communication. Category: Engineering Seniority: Senior Level Tools: Megatron-LM, NVIDIA NeMo, FSDP, Triton, Python, C++, Rust, PyTorch, JAX, DeepSpeed, CUDA Commitment: Full Time Workplace: Hybrid Languages: English