Figma

Figma

Figma provides a collaborative web application for interface design that enables users to seamlessly design, prototype, and gather feedback in real time, while also offering additional offline capabilities through desktop applications for macOS and Win...

Internet Software & Services
1K-5K
Founded 2012
$333M raised

Description

  • Design, build, and productionize machine learning models for search, discovery, ranking, retrieval-augmented generation (RAG), and generative AI features.
  • Build and maintain scalable data pipelines to collect training and evaluation datasets, including annotation systems and human-in-the-loop workflows.
  • Collaborate with AI researchers to refine datasets, evaluation metrics, and model architectures.
  • Work with product engineers to define and deliver AI features across Figma’s platform.
  • Partner with infrastructure engineers to develop and optimize training, inference, monitoring, and deployment systems.
  • Explore new technical ideas and help shape Figma’s long-term AI vision.
  • Collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver ML-driven features and infrastructure.

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.
  • Familiarity with ML libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
  • Experience designing and building scalable data and annotation pipelines.
  • Experience building evaluation systems for AI model quality.
  • Experience mentoring or leading others and contributing to technical excellence and innovation.
  • Familiarity with search relevance, ranking, NLP, or RAG systems is preferred.
  • Experience with AI infrastructure and MLOps, including observability, CI/CD, and automation for ML workflows, is preferred.
  • Experience working on creative or design-focused ML applications is preferred.
  • Knowledge of additional languages such as C++ or Go is a plus.
  • A product mindset with the ability to connect technical work to user outcomes and business impact is preferred.
  • Strong collaboration and communication skills, especially across engineering, product, and research teams is preferred.

Benefits

  • Annual base salary range of $153,000 to $376,000 USD, depending on location and level.
  • Equity as part of the compensation package.
  • Health, dental, and vision insurance.
  • Retirement plan with company contribution.
  • Parental leave and reproductive or family planning support.
  • Mental health and wellness benefits.
  • Generous PTO and company recharge days.
  • Learning and development stipend, work-from-home stipend, and cell phone reimbursement.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Software Engineer, Core Experiences - San Francisco, CA, USA

Speechify 51-250 Internet Software & Services

Speechify is hiring a Senior Software Engineer for its Core Experiences Team to build foundational services and SDKs that power the product experience across platforms.

GCP Java Kotlin
23 minutes ago

Software Engineer, Platform - Malaga, Spain

Speechify 51-250 Internet Software & Services

Speechify is hiring a backend engineer to own and evolve the platform services behind its text-to-speech products in a fully distributed environment.

AWS Azure Docker GCP Kubernetes Microservices Node.js TypeScript
47 minutes ago

Software Engineer, Data Infrastructure & Acquisition - Princeton, NJ, USA

Speechify 51-250 Internet Software & Services

Speechify is hiring a Software Engineer for its AI team’s data function to build and operate data collection and ingestion systems that support large-scale model training and next-generation products.

Bash Docker GCP Linux Python Terraform
49 minutes ago

Software Engineer, Platform - Kyiv, Ukraine

Speechify 51-250 Internet Software & Services

Speechify is seeking a backend engineer for its Platform team to build and maintain core services that power its text-to-speech products and enterprise integrations.

AWS Azure Docker GCP Kubernetes Microservices Node.js REST API TypeScript
55 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers