Software Engineer, Machine Learning

Figma
San Francisco, Canada, United States
Workplace: HybridFull timeUSD 149,000 - 350,000 annuallyFunction: Software EngineeringExperience: 5+ yearsSkills: ["Collaboration","Communication","Leadership"]

Join Figma’s AI team as a Machine Learning Engineer focused on building scalable ML models and data pipelines that power search, discovery, and generative AI features. Collaborate with researchers, engineers, designers, and product managers to deploy high-quality ML systems, while shaping infrastructure for training, inference, and monitoring across the platform. Remote in the US or work from a US hub; contribute to long-term AI strategy.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Figma
Figma
1 year ago

Software Engineer, Machine Learning

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 21 hours agoStatus: Live

Job Summary

Join Figma’s AI team as a Machine Learning Engineer focused on building scalable ML models and data pipelines that power search, discovery, and generative AI features. Collaborate with researchers, engineers, designers, and product managers to deploy high-quality ML systems, while shaping infrastructure for training, inference, and monitoring across the platform. Remote in the US or work from a US hub; contribute to long-term AI strategy.
Location: San Francisco, Canada, United States
Workplace: Hybrid
Employment Type: Full time
Job Function: Software Engineering

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

Pay and Benefits

Salary: USD 149,000 - 350,000 annually
Perks:Health InsuranceDentalVisionEquityLearning BudgetRemote Work

Key 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.
Experience:5+ yearsAIMLMachine learningSoftware engineering
Skills:CollaborationCommunicationLeadership
Languages:English
Tech Stack:PythonPyTorchTensorFlowScikit-learnSparkMLlibXGBoost

Company Brief

Figma
Design collaboration platform for teams that enables interface design, prototyping, and real-time collaboration in the browser. Figma streamlines product design workflows, handoff to developers, and design system management for organizations of all sizes.
Industry: Enterprise Software
Company Size: Enterprise (1,001+ employees)
Growth: Scaleup
Valuation: Decacorn (USD 10B+)
Funding: Series E+
Headquarters: San Francisco, United States
Founded: 2012
WebsiteLinkedIn