Senior Machine Learning Engineer - Photo AI

5 Months ago • 4-8 Years • Research Development

Job Summary

Job Description

Canva's Photo AI team seeks a Senior Machine Learning Engineer to build and improve AI-powered photo editing features. Responsibilities include developing ML models (CNNs, transformers, GANs, diffusion models) using PyTorch (or TensorFlow/JAX), Python, and related libraries. The role involves project ownership, collaborating with frontend/backend teams, staying current with AI advancements, optimizing model efficiency, and working with large datasets on cloud platforms. Candidates should have a Master's (ideally Ph.D.) in Computer Science with relevant publications, strong communication skills, and a demonstrated ability to solve complex problems.
Must have:
  • Master's/PhD in Computer Science (ML focus)
  • Proficiency in PyTorch, Python, Computer Vision
  • Experience with CNNs, Transformers, GANs, Diffusion Models
  • Deep learning at scale and cloud deployment
  • Project ownership and problem-solving skills
Perks:
  • Equity packages
  • Inclusive parental leave
  • Annual Vibe & Thrive allowance
  • Flexible leave options

Job Details

Job Description

Join the team redefining how the world experiences design.

Ahoj, hey, g'day, mabuhay, kia ora, 你好, hallo, vítejte!

Thanks for stopping by. We know job hunting can be a little time consuming and you're probably keen to find out what's on offer, so we'll get straight to the point.

Where and how you can work

Our flagship campus is in Sydney, Australia but Prague is home to part of our European operations. And with that, you'll have choice in how you work, we trust our Canvanauts to choose the balance that empowers them and their team to achieve their goals.

What you’d be doing in this role

As Canva scales change continues to be part of our DNA. But we like to think that's all part of the fun. So this will give you the flavour of the type of things you'll be working on when you start, but this will likely evolve.

At the moment, this role is focused on:

  • Build ML models for AI-based photo editing. This also heavily involves staying up-to-date with the literature and evaluating state-of-the-art models that are applicable for commercial use.
  • Take over project ownership responsibilities and drive projects with specialties like Frontend and Backend to success. This involves ownership of project timeline, and collaboration with Design and Product to achieve the best fitting product for Canva.
  • Stay up-to-date with the very recent advances of AI for photo editing, and you bring new ideas to the team for new AI-based features that our team should build
  • Improve the quality of our existing features, by improving our quality evaluation processes and by bringing new ML ideas to the team
  • Boost the efficiency of our existing features through code optimization, incorporating new models or Python libraries, or by re-evaluating and speeding up our full stack.
  • Collect data that we need for model training or model evaluation or you delegate that to the Data Labelling team.

You're probably a match if

  • You have a Master's (ideally Ph.D.) degree in Computer Science (or similar), focusing on machine learning & published papers in peer-reviewed journals and conferences.
  • You have a solid understanding of computer vision, especially the concepts of convolutional neural networks (CNNs), transformers, generative adversarial networks, and diffusion models.
  • You are proficient with: Deep learning frameworks (Pytorch gives bonus points, alternatively other frameworks like Tensorflow or JAX), Python 3, Numpy, OpenCV, Docker & Kubernetes.
  • You are able to pick up knowledge of the state-of-the-art of a specific ML topic quickly since most projects start with a phase of literature research and communicating the context to the team (across timezones).
  • You have a deep understanding of machine learning at scale, including training sophisticated models on big datasets. You’re experienced in deploying these models on cloud platforms to efficiently serve millions of users.
  • You have a demonstrable sense of ownership and are eager to find and develop novel solutions to challenging problems, such as developing ML models for processing large images in a memory-efficient and runtime-efficient manner.
  • You are an excellent communicator and teammate, keen to work with your colleagues and brainstorm ideas together.


About the team

We are the Photo AI team. We push the boundaries of what is possible in Photo Editing and make a lasting impact on our users' creative journeys. Our mission is to position Canva as a leader in Visual AI by consistently delivering cutting-edge, AI based features that enhance the Photo Editing experience. Our spirit is to set realistic, but crazy-big goals where we dive into the unknown and make new things happen – which is a collaborative effort from all the people in the team. We believe in the power of innovation and high-quality delivery, which is made possible through our ability to focus on what is more impactful for our users and through our ability to invent novel solutions to hard image processing and understanding problems. We are part of the Multimedia Group and together we make Multimedia (photo and video) storytelling quick, simple, and fun for everyone.

What's in it for you?

Achieving our crazy big goals motivates us to work hard - and we do - but you'll experience lots of moments of magic, connectivity and fun woven throughout life at Canva, too. We also offer a range of benefits to set you up for every success in and outside of work.

Here's a taste of what's on offer:

  • Equity packages - we want our success to be yours too
  • Inclusive parental leave policy that supports all parents & carers
  • An annual Vibe & Thrive allowance to support your wellbeing, social connection, office setup & more
  • Flexible leave options that empower you to be a force for good, take time to recharge and supports you personally

Check out lifeatcanva.com for more info.

Other stuff to know

We make hiring decisions based on your experience, skills and passion, as well as how you can enhance Canva and our culture. When you apply, please tell us the pronouns you use and any reasonable adjustments you may need during the interview process.

We celebrate all types of skills and backgrounds at Canva so even if you don’t feel like your skills quite match what’s listed above - we still want to hear from you!

Please note that interviews are conducted virtually.

Similar Jobs

Vigaet - Unreal Engine Cinematic Artist

Vigaet

Bengaluru, Karnataka, India (On-Site)
1 Year ago
Icon Creative - Intermediate Surfacing Artist (On-Site)

Icon Creative

Vancouver, British Columbia, Canada (On-Site)
7 Months ago
Activision - Expert Character Artist - Heads

Activision

Santa Monica, California, United States (On-Site)
2 Months ago
Rigi - Promo Editor

Rigi

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Penrose studios - Director of Marketing

Penrose studios

San Francisco, California, United States (On-Site)
4 Years ago
NVIDIA - Senior Technical Program Manager - Deep Learning Enterprise Server Software

NVIDIA

Taipei City, Taiwan (On-Site)
5 Months ago
PwC - Senior AI Developer - Roma [DIG]

PwC

Rome, Lazio, Italy (On-Site)
9 Months ago
Dream Sports - Senior ML Scientist

Dream Sports

Mumbai, Maharashtra, India (On-Site)
8 Months ago
Zazz - Machine Learning Engineer

Zazz

(Remote)
4 Months ago
Scale AI - Software Engineer, Enterprise AI

Scale AI

New York, New York, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Senior Account Strategist, Mid-Market Sales

Google

New York, New York, United States (On-Site)
2 Months ago
The Walt Disney Company - Senior Software Engineer - Activation

The Walt Disney Company

Santa Monica, California, United States (On-Site)
2 Months ago
Hike - Lead Marketing & PR - Web3

Hike

(Remote)
1 Month ago
kinesthetic games - Animation Director

kinesthetic games

Guildford, England, United Kingdom (Hybrid)
1 Month ago
White board games - 3D Environment Artist (SSR)

White board games

(Remote)
2 Months ago
Global Business Travel - Senior Insights Analyst (Rail)

Global Business Travel

London, England, United Kingdom (On-Site)
1 Month ago
Safehouse - Entry

Safehouse

Tokyo, Japan (On-Site)
2 Days ago
Sprinkler - Consultant - Insights Analyst

Sprinkler

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
Techland - Technical Support Specialist (with English)

Techland

Ostrów Wielkopolski, Greater Poland Voivodeship, Poland (On-Site)
3 Months ago
Scanline VFX - Editor (Trailers)

Scanline VFX

Vancouver, British Columbia, Canada (Hybrid)
9 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Prague, Czechia

bohemia interactive - Technical Designer

bohemia interactive

Prague, Prague, Czechia (On-Site)
5 Months ago
Make - Software Engineering Intern

Make

Prague, Czechia (On-Site)
1 Month ago
bohemia interactive - Lead Character Artist

bohemia interactive

Prague, Prague, Czechia (On-Site)
5 Months ago
Veeam Software - Senior QA Engineer

Veeam Software

Prague, Czechia (Remote)
1 Month ago
Keen Software House - Senior Render Programmer

Keen Software House

Prague, Prague, Czechia (Remote)
4 Months ago
WebTech Corporation - Painter

WebTech Corporation

Pilsen, Plzeň Region, Czechia (On-Site)
3 Weeks ago
virtous games - Junior Dev QA

virtous games

Prague, Prague, Czechia (On-Site)
6 Days ago
Warhorse Studios - DevOps / Infrastructure Engineer

Warhorse Studios

Prague, Prague, Czechia (On-Site)
3 Weeks ago
mad finger games - Build Engineer

mad finger games

Brno, South Moravian Region, Czechia (On-Site)
7 Months ago
Warhorse Studios - Rigger

Warhorse Studios

Prague, Prague, Czechia (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Illumina - Lead Bioinformatics Scientist

Illumina

Cambridge, England, United Kingdom (Hybrid)
2 Weeks ago
easygo - Senior Machine Learning Operations Engineer

easygo

Melbourne, Victoria, Australia (On-Site)
2 Months ago
bytedance - Research Scientist Graduates, Large Language Model (Doubao) - 2025 Start

bytedance

San Jose, California, United States (On-Site)
8 Months ago
NVIDIA - Senior AI-HPC Cluster Engineer

NVIDIA

Santa Clara, California, United States (Hybrid)
3 Months ago
Zazz - Artificial Intelligence Engineer

Zazz

(Remote)
4 Months ago
Valeo - R&D Trainee/Apprentice/VIE

Valeo

Chrzanów, Lesser Poland Voivodeship, Poland (On-Site)
1 Year ago
binance - AI Agent Engineer

binance

Taipei City, Taiwan (On-Site)
2 Months ago
Amazon games  - Senior Software Engineer, Amazon Games AI Research

Amazon games

San Diego, California, United States (On-Site)
6 Months ago
Canonical - Talent Scientist

Canonical

(Remote)
1 Month ago
Trend Micro - (Sr.) Threat Researcher

Trend Micro

Taipei City, Taiwan (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Auckland, Auckland, New Zealand (Remote)

Austin, Texas, United States (Remote)

Sydney, New South Wales, Australia (Remote)

Auckland, Auckland, New Zealand (Remote)

Brisbane, Queensland, Australia (Remote)

Sydney, New South Wales, Australia (Remote)

Austin, Texas, United States (On-Site)

Austin, Texas, United States (Remote)

Austin, Texas, United States (Remote)

View All Jobs

Get notified when new jobs are added by Canva

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug