Figma is seeking to make design accessible to all and is looking for a Machine Learning/AI Engineer to join their growing AI team to drive user productivity, shape new product experiences, and advance the state of AI at Figma.
Requirements
- 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.
- Familiarity with search relevance, ranking, NLP, or RAG systems.
- Experience with AI infrastructure and MLOps, including observability, CI/CD, and automation for ML workflows.
Responsibilities
- 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.
Other
- Ability to work full-time from one of Figma's US hubs or remotely in the United States.
- Must be able to attend in-person onboarding.
- Must keep camera on during video interviews.
- Must be willing to work in a collaborative environment and communicate effectively across functions.
- Must be willing to learn and develop skills as part of Figma's 'Grow as you go' value.
