Figma
Software Engineer - Machine Learning
$153K–$376K
San Francisco, CA • New York, NY • United States +2 · Full-time · Posted Jun 2, 2025
- Location
- San Francisco, CA • New York, NY • United States
Canada
US - Experience
- 5+ years in software engineering, with 3+ years focused on applied machine learning or AI.
About the job
Build and productionize machine-learning models, data pipelines, and AI infrastructure for Figma’s search, discovery, and creativity products.
What you'll do
- Develop ML models and scalable data, annotation, and evaluation pipelines.
- Collaborate across research, product, and infrastructure teams to deliver and improve AI features and systems.
- Build and optimize systems for model training, inference, monitoring, and deployment.
What they ask for
- 5+ years of industry software engineering experience, including 3+ years focused on applied machine learning or AI.
- End-to-end ML model development experience, including training, evaluation, deployment, and monitoring.
- Proficiency in Python and familiarity with ML libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
- Experience building scalable data and annotation pipelines and AI model evaluation systems.
Nice to have
- Familiarity with search relevance, ranking, NLP, or RAG systems; AI infrastructure and MLOps.
- Experience with creative or design-focused ML applications; C++ or Go.
Pay
$153,000 — $376,000 USD (base salary)
Tools
Python, PyTorch, TensorFlow, Scikit-learn, Spark MLlib, XGBoost, C++, Go
AI focus
Applied ML and AI for search, discovery, ranking, retrieval-augmented generation (RAG), and generative creativity features; includes dataset development, evaluation, and ML infrastructure.
Model-written summary · the original posting is authoritative