Senior Machine Learning Systems Engineer

2 days ago

Beijing, Beijing, China Canva Full time $120,000 - $240,000 per year

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.

Job Description
About the Role/Specialty
As a Senior Machine Learning Systems 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 and inference.
  • 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.
  • You'll evaluate and apply best practices for distributed training using industry-leading frameworks.
  • You'll dive deep into low-level optimization, including custom CUDA or Triton kernels.
  • You'll debug, profile, and fine-tune training workflows to unlock new levels of scalability.

Qualifications
What we're looking for
We're looking for a systems-first engineer who thrives in fast-paced, high-impact environments. You're deeply familiar with distributed model training at scale and understand the nuances of optimizing compute at every level of the stack. You're excited by challenges that stretch current boundaries, and you're a strong collaborator who communicates clearly across domains.

  • Strong background in LLMs, multimodal AI, or diffusion models.
  • Proficiency in Python. Familiarity with a system programming language (e.g. C++ or Rust) is a plus.
  • Deep knowledge of PyTorch or JAX as well as libraries such as Megatron-LM, NeMo, or DeepSpeed.
  • Familiarity with common optimization techniques such as FSDP/ZeRO, gradient checkpointing, or low-precision data types.
  • Hands-on experience writing custom GPU kernels in CUDA or Triton.
  • Excellent communication and problem-solving skills, incl. full proficiency in English.

  • Senior Machine Learning Systems Engineer

    2 days ago

    Beijing, Beijing, China Canva Full time CN¥300,000 - CN¥900,000 per year

    Company Description About the Group/TeamWe'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...

  • Machine Learning Engineer

    6 days ago

    Beijing, Beijing, China Conviva Full time CN¥100,000 - CN¥120,000 per year

    Conviva is the intelligence layer for digital businesses, turning every consumer interaction into outcome-based intelligence—linking engagement patterns across AI agents, apps, websites, and streaming video to real results You will delve into a diverse array of ML technologies to drive innovative customer experiences. Collaborating closely with vertical...

  • Machine Learning Engineer

    2 days ago

    Beijing, Beijing, China Conviva Full time CN¥120,000 - CN¥240,000 per year

    Conviva is the intelligence layer for digital businesses, turning every consumer interaction into outcome-based intelligence—linking engagement patterns across AI agents, apps, websites, and streaming video to real results such as purchases, bookings, and resolved support requests. Powered by its patented Time-State Technology, the Conviva Operational Data...

  • Senior Machine Learning Engineer

    6 days ago

    Beijing, Beijing, China Coupang Internal Full time CN¥300,000 - CN¥600,000 per year

    Please complete the attached Internal Transfer Request Form and submit.   Please make sure to apply with your Coupang e-mail address.   Company Introduction  We exist to wow our customers. We know we're doing the right thing when we hear our customers say, "How did we ever live without Coupang?" Born out of an obsession to make shopping,...

  • Senior Machine Learning Engineer

    6 hours ago

    Beijing, Beijing, China Gibbs & Cox Australia Full time

    Company DescriptionAbout Grab and Our WorkplaceGrab is Southeast Asia's leading superapp. From getting your favourite meals delivered to helping you manage your finances and getting around town hassle-free, we've got your back with everything. In Grab, purpose gives us joy and habits build excellence, while harnessing the power of Technology and AI to...

  • Senior Machine Learning Engineer

    6 hours ago

    Beijing, Beijing, China Grab Full time

    Company Description About Grab and Our WorkplaceGrab is Southeast Asia's leading superapp. From getting your favourite meals delivered to helping you manage your finances and getting around town hassle-free, we've got your back with everything. In Grab, purpose gives us joy and habits build excellence, while harnessing the power of Technology and AI to...

  • Senior Staff/Staff Machine Learning Engineer

    6 days ago

    Beijing, Beijing, China Coupand Full time CN¥200,000 - CN¥600,000 per year

    Senior Staff/Staff Machine Learning Engineer-LLMCompany IntroductionWe exist to wow our customers. We know we're doing the right thing when we hear our customers say, "How did we ever live without Coupang?" Born out of an obsession to make shopping, eating, and living easier than ever, we are collectively disrupting the multi-billion-dollar commerce industry...

  • Machine Learning Engineer

    4 days ago

    Beijing, Beijing, China Grab Full time

    Company DescriptionAbout Grab and Our WorkplaceGrab is Southeast Asia's leading superapp. From getting your favourite meals delivered to helping you manage your finances and getting around town hassle-free, we've got your back with everything. In Grab, purpose gives us joy and habits build excellence, while harnessing the power of Technology and AI to...

  • Machine Learning Engineer

    6 days ago

    Beijing, Beijing, China Grab Full time CN¥120,000 - CN¥240,000 per year

    Company Description About Grab and Our WorkplaceGrab is Southeast Asia's leading superapp. From getting your favourite meals delivered to helping you manage your finances and getting around town hassle-free, we've got your back with everything. In Grab, purpose gives us joy and habits build excellence, while harnessing the power of Technology and AI to...

  • Computer Vision/Machine Learning Engineer

    4 days ago

    Beijing, Beijing, China Apple Full time CN¥300,000 - CN¥600,000 per year

    If you are passionate about advancing multi-modal understanding, building models that bridge text and visual perception, and shaping the next generation of intelligent on-device experiences, Apple is the right place for you. We are looking for engineers who combine technical depth, curiosity, and creativity to push the boundaries of what machine learning can...