ML Engineer - AI Platform
FigmaJob Summary
Figma is seeking AI engineers with experience in building data pipelines and evaluation systems for AI models. The role involves collaborating with researchers and product engineers to enhance Figma's AI features, build new capabilities, and advance AI technology for creativity. Key responsibilities include developing data pipelines, annotation pipelines, working closely with AI teams on requirements, improving datasets and metrics, infrastructure development, and exploring innovative solutions. We value expertise in AI/ML infrastructure, software development (especially Python), collaboration with researchers, dataset improvement, problem-solving, and cross-functional communication. At Figma, we encourage growth and learning, so even if your experience doesn't perfectly align, we welcome you to apply.
We’re looking for AI engineers who have experience building data pipelines to collect high quality data, and evaluation systems to evaluate AI models. You will be collaborating with AI researchers and product engineers closely to improve Figma’s AI features, build new capabilities and advance the state of the art in AI for creativity.
What you’ll do at Figma:
Build data pipelines to collect high quality training and evaluation data for developing AI models.
Build annotation pipelines to collect ground truth using humans and AI models in the loop.
Collaborate closely with AI product engineers to define requirements for AI features.
Collaborate closely with AI researchers to identify failure cases, improve data sets, evaluation metrics and hill climb quality of AI models.
Work closely with AI infra engineers to build the right infrastructure for AI model development and inference.
Explore the boundaries of what is possible with the current technology set and experiment with novel ideas.
We'd love to hear from you if you have:
Expertise building infrastructure for AI/ML model development in industry.
Experience in software development in Python. Experience in languages like C/C++ is nice to have but not required.
Expertise working closely with AI researchers on applied AI projects to build high quality models in a fast paced environment.
Expertise improving datasets and evaluation metrics for applied ML projects.
Experience solving sophisticated problems and comparing alternative solutions, trade-offs, and diverse points of view to determine a path forward
Experience communicating and working across functions to drive solutions.
At Figma, one of our values is Grow as you go. We believe in hiring smart, curious people who are excited to learn and develop their skills. If you’re excited about this role but your past experience doesn’t align perfectly with the points outlined in the job description, we encourage you to apply anyways. You may be just the right candidate for this or other roles.
