Software Engineer - Machine LearningActive$376K

The opportunity

Figma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma’s platform helps teams bring ideas to life—whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI.

What you'll do

  • 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.

What they're looking for

  • + 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.