Software Engineer, Machine Learning
Core
Building intelligent search systems, scalable data pipelines, and AI-powered creativity tools to drive user productivity and advance the state of AI at Figma.
Role type
Senior IC machine learning engineer (applied AI)
Builds
Production ML models, scalable data pipelines, and AI features for design and collaboration tools
Domain
SaaS / Design & Collaboration / Applied Machine Learning
Deliverable
production ML models | product features | infrastructure
Required skills
End-to-end ML model development, Python, PyTorch/TensorFlow/Scikit-learn/Spark MLlib/XGBoost, scalable data and annotation pipelines, evaluation systems for AI model quality, mentoring
Preferred skills
Search relevance/ranking/NLP/RAG, AI infrastructure/MLOps, creative or design-focused ML applications, C++/Go, product mindset
Technologies
PyTorch, TensorFlow, Scikit-learn, Spark MLlib, XGBoost
Responsibilities
Design, build, and productionize ML models for Search, Discovery, Ranking, RAG, and generative AI; Build and maintain scalable data pipelines for training and evaluation datasets; Collaborate with researchers to iterate on datasets, metrics, and architectures; Work with product engineers to deliver AI features; Partner with infrastructure engineers to optimize training, inference, and deployment systems; Explore new technical ideas to shape long-term AI vision
Seniority
Senior, hands-on IC