Senior Machine Learning Engineer - Photo AI

15 Hours ago • 4-8 Years • Artificial Intelligence • Undisclosed

About the job

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 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.

View Full Job Description

Add your resume

80%

Upload your resume, increase your shortlisting chances by 80%

About The Company

Sydney, New South Wales, Australia (Remote)

Vienna, Vienna, Austria (Remote)

Sydney, New South Wales, Australia (Remote)

Sydney, New South Wales, Australia (Remote)

Sydney, New South Wales, Australia (Remote)

Sydney, New South Wales, Australia (Remote)

Perth, Western Australia, Australia (Remote)

Paris, Île-de-France, France (Remote)

Melbourne, Victoria, Australia (Remote)

View All Jobs

Get notified when new jobs are added by Canva

Similar Jobs

ByteDance - Software Engineer, Inference

ByteDance, United States (On-Site)

Playrix - Feature Owner (LiveOps)

Playrix, Ukraine (Remote)

G2 - Data Scientist - II

G2, India (Hybrid)

Meta - Software Engineer, Machine Learning

Meta, United States (On-Site)

ASSIST Software - AI Engineer

ASSIST Software, Romania (Remote)

Arrise Solutions (India)   - Lead ML Engineer

Arrise Solutions (India) , India (On-Site)

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Playrix - Game Designer

Playrix, (Remote)

Playrix - Game Designer

Playrix, Cyprus (Remote)

Tech Mahindra - Computational Linguist

Tech Mahindra, India (On-Site)

Paypal - Staff Machine Learning Engineer

Paypal, United States (Hybrid)

Inizio Advisory - Data Science Manager

Inizio Advisory, India (Hybrid)

ByteDance - Machine Learning Research Scientist, AI for Science

ByteDance, United States (On-Site)

Paypal - Machine Learning Manager

Paypal, United States (Hybrid)

G2 - Data Scientist - II

G2, India (Hybrid)

Playkot - Senior 2D Artist (Generalist)

Playkot, Serbia (Remote)

Get notifed when new similar jobs are uploaded

Jobs in Prague, Czechia

Make - Data Engineer

Make, Czechia (Hybrid)

NXP - Student - Ethernet SW developper

NXP, Czechia (On-Site)

MADFINGER Games - Senior UI Programmer

MADFINGER Games, Czechia (On-Site)

Virtuos - Engine programmer - Unreal Engine

Virtuos, Czechia (Hybrid)

PwC - CRM Leader

PwC, Czechia (On-Site)

Microsoft - Software Engineering

Microsoft, Czechia (On-Site)

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Get notifed when new similar jobs are uploaded