Senior Machine Learning Engineer - Photo AI

3 Months 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

Playtika - Experienced Data Scientist

Playtika

Israel (On-Site)
2 Months ago
Playrix - Game Designer

Playrix

Ireland (Remote)
6 Months ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Astana, Astana, Kazakhstan (Remote)
3 Months ago
Playrix - Game Designer

Playrix

Ukraine (Remote)
6 Months ago
Ello - Tech Lead, GenAI & Machine Learning

Ello

San Francisco, California, United States (On-Site)
2 Weeks ago
NVIDIA - Senior Computer Architect - Deep Learning

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago
Meta - Research Scientist Intern, Smart Glasses in Wearables AI (PhD)

Meta

Redmond, Washington, United States (On-Site)
5 Months ago
ByteDance - Student Researcher Intern (Edge Research Project for General Intelligence)

ByteDance

San Jose, California, United States (On-Site)
2 Weeks ago
Zoox - Senior/Staff Machine Learning Engineer - Prediction & Behavior ML

Zoox

Foster City, California, United States (Hybrid)
6 Months ago
Google - Physical Design Engineer, TPU

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Ello - Tech Lead, GenAI & Machine Learning

Ello

San Francisco, California, United States (On-Site)
2 Weeks ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Tbilisi, Tbilisi, Georgia (Remote)
3 Months ago
Vigaet - Machine Learning Engineer-Internship

Vigaet

Bengaluru, Karnataka, India (On-Site)
1 Year ago
CD PROJEKT RED - Data Scientist

CD PROJEKT RED

Warsaw, Masovian Voivodeship, Poland (On-Site)
1 Day ago
Tencent - Senior Staff Researcher

Tencent

Palo Alto, California, United States (On-Site)
6 Months ago
Monzo - Senior Machine Learning Scientist

Monzo

London, England, United Kingdom (Hybrid)
6 Hours ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Video Generative Model)

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
ByteDance - Senior AI Space Interaction Industrial Designer

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Playrix - Feature Owner (LiveOps)

Playrix

Almaty, Almaty Region, Kazakhstan (Remote)
6 Months ago
DISCO - Engineering Manager , Feature

DISCO

Gurugram, Haryana, India (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Prague, Czechia

Wrike - Regional Marketing Manager (French)

Wrike

Prague, Prague, Czechia (Hybrid)
1 Day ago
Ness Digital - Data Solutions Program Manager

Ness Digital

Prague, Czechia (On-Site)
1 Month ago
PwC - Data Engineer - Financial Crime team

PwC

Prague, Prague, Czechia (On-Site)
7 Months ago
PwC - Senior BI/Data analyst – Technology Consulting

PwC

Prague, Prague, Czechia (On-Site)
6 Months ago
VECTOR NORTH - [Vector North] Senior Game Designer

VECTOR NORTH

Prague, Prague, Czechia (Hybrid)
5 Months ago
Keen Software House - Senior Render Programmer

Keen Software House

Prague, Prague, Czechia (Remote)
2 Months ago
Bohemia Interactive - IT Technician

Bohemia Interactive

Mníšek, Liberec Region, Czechia (On-Site)
2 Weeks ago
Enverus - Senior Site Reliability Engineer

Enverus

Brno, South Moravian Region, Czechia (Hybrid)
23 Hours ago
Virtuos - Software Engineer

Virtuos

Czechia (Hybrid)
2 Weeks ago
VECTOR NORTH - [Vector North] Senior Gameplay Programmer

VECTOR NORTH

Prague, Prague, Czechia (Hybrid)
6 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Garena - AI Image Creation Collaborator

Garena

Hanoi, Hanoi, Vietnam (On-Site)
2 Weeks ago
Google - Research Scientist, Google Cloud AI

Google

Kirkland, Washington, United States (On-Site)
1 Week ago
Keywords Studios - AI - Research Associate (Prompts)

Keywords Studios

Silesian Voivodeship, Poland (On-Site)
1 Month ago
Netflix - Product Manager, ML Platform: Training

Netflix

Los Gatos, California, United States (Hybrid)
6 Months ago
NVIDIA - Research Scientist, Deep Learning and Computer Vision

NVIDIA

Hsinchu, Hsinchu City, Taiwan (On-Site)
2 Months ago
Vigaet - Internship- AI Engineer

Vigaet

Bengaluru, Karnataka, India (On-Site)
1 Year ago
Google - Cloud Engineer II, AI/ML, Professional Services

Google

Mexico City, Mexico City, Mexico (On-Site)
1 Week ago
Microsoft - Member of Technical Staff - AI Multimodal

Microsoft

Zürich, Zurich, Switzerland (On-Site)
2 Weeks ago
Google - Customer Engineer, AI Infrastructure

Google

Seattle, Washington, United States (On-Site)
2 Weeks ago
Corsair - Senior Manager, AI & Data

Corsair

Munich, Bavaria, Germany (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded