Latest AI JobsBrowse Jobs
Figma logo

Software Engineer - Machine Learning

FigmaSan Francisco, CA • New York, NY • United StatesEngineering
Apply on company site
Salary
Not listed
Job type
Full-time
Posted
2 Jun 2025
Status
Open · checked 7d ago

About this role

Figma is seeking a Software Engineer specializing in Machine Learning to join the artificial intelligence team based in San Francisco, New York, or remotely in the United States. This dynamic position suits a professional with a strong technical background who is eager to build advanced machine learning models and scalable infrastructure that power collaborative design tools.

Role Overview

Figma is looking for a versatile and experienced Machine Learning Engineer to operate at the intersection of applied machine learning, infrastructure, and product innovation. The selected candidate will build intelligent search systems, craft scalable data pipelines, and enhance AI-powered creativity tools to drive user productivity. Working closely with engineers, researchers, designers, and product managers across multiple teams, this role delivers impactful ML-driven features and foundational systems.

Key Responsibilities

Design, build, and productionize machine learning models for search, discovery, ranking, retrieval-augmented generation (RAG), and generative AI features.
Construct and maintain scalable data pipelines to gather high-quality training and evaluation datasets, including annotation systems and human-in-the-loop workflows.
Partner with AI researchers to iterate on datasets, evaluation metrics, and model architectures to enhance quality and relevance.
Collaborate with product engineers to define and deliver impactful artificial intelligence features throughout the platform.
Team up with infrastructure engineers to optimize systems dedicated to training, inference, monitoring, and deployment.
Explore new technical concepts to help shape the long-term artificial intelligence vision for the company.

Requirements

Minimum of five years of industry experience in software engineering, with at least three years specifically focused on applied machine learning or artificial intelligence.
Strong expertise in end-to-end machine learning model development, covering training, evaluation, deployment, and monitoring.
Proficiency in Python alongside familiarity with machine learning libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
Proven background in designing and building scalable data and annotation pipelines alongside evaluation systems for model quality.
Experience mentoring or leading other team members while contributing to a culture of technical excellence and innovation.

Nice to Have

Familiarity with search relevance, ranking, natural language processing, or RAG systems.
Experience with artificial intelligence infrastructure and MLOps, including observability, CI/CD, and workflow automation.
Background working on creative or design-focused machine learning applications.
Knowledge of additional programming languages like C++ or Go.
A product mindset connecting technical output to user outcomes and business impact, alongside strong cross-functional communication skills.

Skills and Tools

Python, PyTorch, TensorFlow, Scikit-learn, Spark MLlib, XGBoost, C++, Go
Machine learning operations, MLOps, CI/CD, observability
Natural language processing, RAG, search, ranking, retrieval, generative AI
Scalable data pipelines, annotation systems, evaluation metrics

Salary and Benefits

Annual base salary range between $153,000 and $376,000 USD, with job level and actual compensation determined by qualifications, skills, experience, and location.
Equity offerings and a comprehensive benefits package including health, dental, and vision coverage.
Retirement benefits with company contributions, parental leave, family planning support, and mental health resources.
Paid time off, flexible paid sick leave, flexible paid time off for exempt employees, and paid holidays.
Additional perks such as company recharge days, cell phone and home internet reimbursements, and lifestyle spending accounts.

About the Company

Figma provides a platform that helps teams bring ideas to life through brainstorming, prototyping, translating designs into code, and real-time collaboration. Guided by values such as growing as you go, the organization fosters an inclusive and equal opportunity workplace that celebrates diverse thoughts and backgrounds.

How to Apply

Interested candidates should submit their application directly through the official Figma careers page.

Skills and tech stack

PythonPyTorchTensorFlowSparkC++NLPRAGMLOpsFigma

Questions candidates ask

Where is this position located?

This full-time role can be performed from Figma hubs in San Francisco, CA or New York, NY, or remotely from anywhere within the United States.

What level of experience is required for this role?

Candidates must have 5+ years of industry experience in software engineering, with at least 3+ years specifically focused on applied machine learning or AI.

What is the annual base salary range for the Machine Learning Engineer position?

The annual base salary range is $153,000 to $376,000 USD, with the exact compensation determined by objective factors like skills, experience, and location.

What programming languages and tools should candidates know?

Proficiency in Python is required, along with familiarity with ML libraries like PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost. Knowledge of C++ or Go is considered a plus.

What benefits are provided by Figma?

Figma offers equity, health, dental, and vision coverage, retirement benefits with company matching, parental leave, mental health benefits, paid time off, and reimbursements for cell phones and home internet.

Ready to apply?

The application is on the official Figma careers page.

Apply on company site