Software Engineer, Machine Learning Jobs

Figma

Location:
United States, San Francisco ▼
New York

Category:
IT - Software Development

Contract Type:
Not provided

Salary:

149000.00 - 350000.00 USD / Year

Save Job

Apply Position

Job Description:

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma. You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure. This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities.

Job Responsibility:

Design, build, and productionize ML models for Search, Discovery, Ranking, Retrieval-Augmented Generation (RAG), and generative AI features
Build and maintain scalable data pipelines to collect high-quality training and evaluation datasets, including annotation systems and human-in-the-loop workflows
Collaborate with AI researchers to iterate on datasets, evaluation metrics, and model architectures to improve quality and relevance
Work with product engineers to define and deliver impactful AI features across Figma’s platform
Partner with infrastructure engineers to develop and optimize systems for training, inference, monitoring, and deployment
Explore new ideas at the edge of what’s technically possible and help shape the long-term AI vision at Figma

Requirements:

5+ years of industry experience in software engineering
3+ years focused on applied machine learning or AI
Strong experience with end-to-end ML model development, including training, evaluation, deployment, and monitoring
Proficiency in Python and familiarity with ML libraries like PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost
Experience designing and building scalable data and annotation pipelines, as well as evaluation systems for AI model quality
Experience mentoring or leading others and contributing to a culture of technical excellence and innovation

Nice to have:

Familiarity with search relevance, ranking, NLP, or RAG systems
Experience with AI infrastructure and MLOps, including observability, CI/CD, and automation for ML workflows
Experience working on creative or design-focused ML applications
Knowledge of additional languages such as C++ or Go
A product mindset with the ability to tie technical work to user outcomes and business impact
Strong collaboration and communication skills, especially when working across functions (engineering, product, research)

What we offer:

equity
health, dental & vision benefits
retirement with company contribution
parental leave & reproductive or family planning support
mental health & wellness benefits
generous PTO
company recharge days
a learning & development stipend
a work from home stipend
cell phone reimbursement
annual bonus plan for eligible non-sales roles

Additional Information:

Job Posted:
December 08, 2025

Employment Type:

Fulltime

Work Type:

On-site work

View All Jobs In This Company

Job Link Share:

Software Engineer, Machine Learning