ML Engineer — Design AI

Best AI Tools Wiki

San Francisco (CA)

Hybrid

USD 150,000 - 230,000

Full time

14 days+

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Equity grants
Remote flexibility
Wellness stipend
Parental leave

Job summary

Figma is seeking an ML Engineer to build AI-powered design tools, focusing on layout generation, design-to-code conversion, intelligent auto-layout, and generative design features that help millions of designers work faster.

The role requires 4+ years ML engineering experience, strong Python and TypeScript skills, and a track record deploying ML models in production, with an understanding of design systems and UI/UX principles.

Qualifications

  • 4+ years ML engineering experience.
  • Experience with computer vision or generative models.
  • Strong skills in Python and TypeScript.
  • Understanding of design systems and UI/UX principles.
  • Experience deploying ML models in production.

Responsibilities

  • Build AI-powered design tools and features (layout generation, design-to-code conversion, auto-layout, generative design).
  • Collaborate with design/product teams to ship features that scale for millions of designers.

Skills

Python
TypeScript
Production ML
UI/UX principles

Tools

WebAssembly

Job description

Figma is seeking an ML Engineer to build AI-powered design tools. You will work on layout generation, design-to-code conversion, intelligent auto-layout, and generative design features that help millions of designers work faster.

Requirements
  • 4+ years of ML engineering experience
  • Experience with computer vision or generative models
  • Strong skills in Python and TypeScript
  • Understanding of design systems and UI/UX principles
  • Experience deploying ML models in production
Nice to Have
  • Experience with design tools or creative software
  • Background in layout or document understanding
  • Experience with multimodal models
  • Familiarity with WebAssembly or canvas rendering

Equity grants

Remote flexibility

Wellness stipend

Parental leave

Skills

This listing was posted on 2026-01-15 and is almost certainly closed. It is kept for reference — check the employer’s own careers page for current openings.

We don’t have an application link for this role.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

ML Engineer for AI Design Tools — Remote, Equity & Wellness
ML Engineer for AI Design Tools — Remote, Equity & Wellness

Best AI Tools Wiki • San Francisco (CA)

Hybrid
USD 150,000 - 230,000
Equity grants
Remote flexibility
Wellness stipend
+1
F Software Engineer - Machine Learning Figma via Greenhouse San Francisco, CA • New York, NY • United States Engineering View role
F Software Engineer - Machine Learning Figma via Greenhouse San Francisco, CA • New York, NY • United States Engineering View role

Nubeero Limited • San Francisco (CA)

On-site
USD 153,000 - 376,000
Senior ML Engineer - AI for Creative Design (Remote)
Senior ML Engineer - AI for Creative Design (Remote)

Figma • San Francisco (CA)

Remote
USD 153,000 - 376,000
Health, dental & vision insurance
Retirement with company contribution
Parental leave
+1
AI Applied Scientist
AI Applied Scientist

Figma • New York (NY)

On-site
USD 149,000 - 350,000
Health insurance
Retirement with company contribution
Generous PTO
+1
AI Applied Scientist
AI Applied Scientist

The available sources do not contain information about the company name for rounx.com. • San Francisco (CA)

On-site
USD 149,000 - 350,000
Health, dental & vision insurance
Retirement with company contribution
Generous PTO
+2
Remote ML Engineer – Search, NLP & GenAI
Remote ML Engineer – Search, NLP & GenAI

Figma • San Francisco (CA), New York (NY)

On-site
USD 149,000 - 350,000
Health, dental & vision benefits
Retirement with company contribution
Parental leave & family planning support
+7
AI Applied Scientist
AI Applied Scientist

Figma • San Francisco (CA)

On-site
USD 149,000 - 350,000
Health, dental & vision benefits
Retirement plan with company contribution
Generous PTO and recharge days
+2
Generative AI Scientist — Applied ML for Design
Generative AI Scientist — Applied ML for Design

Figma • San Francisco (CA)

Remote
USD 149,000 - 350,000
Health, dental & vision insurance
Retirement with company contribution
Generous PTO
+2
F AI Applied Scientist Figma via Greenhouse San Francisco, CA • New York, NY • United States Engineering View role
F AI Applied Scientist Figma via Greenhouse San Francisco, CA • New York, NY • United States Engineering View role

Nubeero Limited • San Francisco (CA)

Hybrid
USD 153,000 - 376,000
Equity
Health insurance
Dental coverage
+3
Software Engineer, Machine Learning
Software Engineer, Machine Learning

Figma • San Francisco (CA), New York (NY)

On-site
USD 149,000 - 350,000
Health, dental & vision benefits
Retirement with company contribution
Parental leave & family planning support
+7