Software Engineer - Machine Learning
Core
Design, build, and productionize ML models for Search, Discovery, Ranking, RAG, and generative AI features while building scalable data pipelines.
Role type
Senior IC machine learning engineer (applied AI)
Builds
Intelligent search systems, scalable data pipelines, and AI-powered creativity tools for Figma's design platform.
Domain
Design software / Applied Machine Learning
Deliverable
production ML models
Required skills
End-to-end ML model development, Python, PyTorch, TensorFlow, Spark MLlib, XGBoost, scalable data pipeline design, annotation systems, human-in-the-loop workflows, model evaluation, mentoring
Preferred skills
Search relevance, ranking, NLP, RAG systems, AI infrastructure, MLOps, observability, CI/CD, automation, creative/design-focused ML applications, C++, Go
Technologies
PyTorch, TensorFlow, Scikit-learn, Spark MLlib, XGBoost
Responsibilities
Design and productionize ML models for search and generative AI; build and maintain scalable data pipelines; collaborate with researchers to iterate on datasets and metrics; partner with product engineers to deliver AI features; optimize training and inference systems.
Seniority
Senior, hands-on IC