# Software Engineer, Machine Learning

**Company:** [Figma](https://hotfix.jobs/companies/figma)
**Location:** Remote
**Role:** ML Engineering
**Salary:** $149k – $350k/yr
**Experience:** 5+ years
**Skills:** Python, PyTorch, TensorFlow, scikit-learn, Spark Mllib, Xgboost, RAG, MLOps, NLP, Kubernetes
**Posted:** 2025-11-19

> Build and productionize ML models for search, RAG, and generative AI features at Figma. Requires 5+ years software engineering with 3+ years in applied ML, Python proficiency, and experience with scalable data pipelines.

## Job Description

## What you’ll do at Figma:
- 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.

## We’d love to hear from you if you have:
- 5+ years of industry experience in software engineering, with 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.

## While not required, it’s an added plus if you also 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.

## Similar roles

- [Software Engineer, AI Platforms](https://hotfix.jobs/jobs/95b79c88-a7b8-469e-af7a-562a42879ba8) - Figma - Remote - $149k – $350k/yr
- [Perception & Fusion Engineer](https://hotfix.jobs/jobs/8655c0c2-c78a-420d-b397-b01bb4bef878) - Applied Intuition - Arlington, VA - $150k – $220k/yr
- [Applied ML Engineer](https://hotfix.jobs/jobs/3016b55f-07d3-486b-ba7c-0c68f9819cd0) - Deepgram - Remote - $150k – $220k/yr
- [Backend Engineer](https://hotfix.jobs/jobs/ab7e0e7b-91ae-4435-ae9f-df959e7843f1) - Deepgram - Remote - $150k – $220k/yr
- [Research Engineer, Post-Training](https://hotfix.jobs/jobs/09d480a0-f2fe-4418-8549-14c5c9fa2aa0) - Distyl AI - San Francisco, CA - $150k – $250k/yr

**Apply:** https://hotfix.jobs/jobs/513fb513-4030-4fb0-a8af-b7064e2b9e63
**Canonical:** https://hotfix.jobs/jobs/513fb513-4030-4fb0-a8af-b7064e2b9e63