Senior Machine Learning Engineer - Photo AI

3 Weeks ago • 4-8 Years • Artificial Intelligence

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 - Machine Learning Engineer-Internship

Vigaet

Bengaluru, Karnataka, India (On-Site)
10 Months ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Astana, Astana, Kazakhstan (Remote)
1 Month ago
Madison Logic - Sr. Data Scientist

Madison Logic

Pune, Maharashtra, India (On-Site)
3 Months ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Tbilisi, Tbilisi, Georgia (Remote)
1 Month ago
Tech Mahindra - Computational Linguist

Tech Mahindra

Hyderabad, Telangana, India (On-Site)
5 Months ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Sausalito, California, United States (On-Site)
3 Months ago
Coursera - AI Specialist

Coursera

India (Remote)
2 Weeks ago
Salesforce - 2025 PhD Intern - AI Research, Singapore

Salesforce

Singapore, Singapore (On-Site)
4 Months ago
MAINEX ACADEMY - Genarative AI Artist

MAINEX ACADEMY

Chennai, Tamil Nadu, India (Hybrid)
5 Months ago
Blizzard Entertainment - Senior Data Scientist, Computer Graphics

Blizzard Entertainment

Irvine, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Playkot - Senior 2D Artist (Generalist)

Playkot

Belgrade, Serbia (Remote)
1 Month ago
ByteDance - Machine Learning Engineer - Global Payment - 2025 Start

ByteDance

Singapore (On-Site)
1 Month ago
Zoox - Collision Avoidance System, Machine Learning Internship/Co-op

Zoox

Foster City, California, United States (On-Site)
4 Months ago
Playrix - Game Designer

Playrix

(Remote)
4 Months ago
LeoVegas - Data Scientist - Sportsbook

LeoVegas

Stockholm, Stockholm County, Sweden (Hybrid)
2 Months ago
NXP - Test infrastructure developer intern

NXP

Roznov, Neamț County, Romania (On-Site)
4 Months ago
NVIDIA - Senior ASIC Verification Engineer, Coherent High Speed Interconnect

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
Luxoft - Senior GPU Compiler Software Development Engineer

Luxoft

Italy, New York, United States (Remote)
2 Months ago
Ello - Tech Lead, Machine Learning

Ello

Canada (On-Site)
3 Months ago
Playrix - Feature Owner (LiveOps)

Playrix

Portugal (Remote)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Prague, Czechia

Keen Software House - Senior Gameplay Programmer

Keen Software House

Prague, Prague, Czechia (Remote)
2 Weeks ago
Microsoft - Technical Program Manager

Microsoft

Prague, Prague, Czechia (Remote)
1 Month ago
Bohemia Interactive - QA Automation Engineer - DayZ

Bohemia Interactive

Prague, Prague, Czechia (On-Site)
3 Weeks ago
Make - E-learning Content Developer

Make

Prague, Czechia (Hybrid)
3 Months ago
Bohemia Interactive - Lead Programmer

Bohemia Interactive

Prague, Prague, Czechia (On-Site)
3 Months ago
PwC - Advokát/koncipient/ka do týmu daňových litigací

PwC

Prague, Prague, Czechia (On-Site)
4 Months ago
NXP - Sr Director of Security Operations

NXP

Brno, South Moravian Region, Czechia (On-Site)
4 Months ago
PwC - Data engineer - Technology consulting

PwC

Prague, Prague, Czechia (On-Site)
4 Months ago
Visa - Consult & Analytics Graduate

Visa

Prague, Czechia (On-Site)
4 Months ago
Bohemia Interactive - Designer - Character and Weapon

Bohemia Interactive

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

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Blue Yonder - AI/ML Penetration Tester Consultant

Blue Yonder

Hyderabad, Telangana, India (On-Site)
4 Months ago
Riot Games - Senior Data Scientist - Singapore Efficiency Team

Riot Games

Singapore (On-Site)
3 Weeks ago
Zazz - Machine Learning Engineer

Zazz

(Remote)
5 Days ago
Krafton  - Deep Learning Strategy & Operations Associate

Krafton

Seoul, South Korea (On-Site)
5 Days ago
Keywords Studios (Player Support) - AI Engineer (AI-Powered Agents)

Keywords Studios (Player Support)

Maharashtra, India (Hybrid)
2 Months ago
The Walt Disney Company - Principal Machine Learning Engineer, Research - Ad Platforms

The Walt Disney Company

Santa Monica, California, United States (On-Site)
3 Months ago
Unity - Senior Data Scientist, iAds

Unity

Helsinki, Uusimaa, Finland (On-Site)
6 Months ago
The Walt Disney Company - Principal Machine Learning Engineer, Research - Ad Platforms

The Walt Disney Company

Seattle, Washington, United States (On-Site)
3 Months ago
NVIDIA - Senior AI-HPC Storage Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Sydney, New South Wales, Australia (Remote)

San Francisco, California, United States (Remote)

Perth, Western Australia, Australia (Remote)

Melbourne, Victoria, Australia (Remote)

Brisbane, Queensland, Australia (Remote)

Auckland, Auckland, New Zealand (Remote)

Sydney, New South Wales, Australia (Remote)

Melbourne, Victoria, Australia (Remote)

Auckland, Auckland, New Zealand (Remote)

Sydney, New South Wales, Australia (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