Figma trains AI Search models in 5 months using Amazon SageMaker AI
FigmaAmazon SageMaker AI · Amazon SageMaker Model Training · Amazon SageMaker Pipelines +4Design platform Figma built its AI infrastructure on Amazon SageMaker AI, using Amazon EMR with Apache Spark to process billions of design elements and the custom FigmaStep framework on SageMaker Pipelines to orchestrate training. Figma trained and deployed the models behind its AI Search feature within five months, launching it at Config 2024, and ran more than 10,000 SageMaker AI training jobs in 2025 to support features including the AI prompt-to-app tool Figma Make.
Overview
Design platform Figma built its AI infrastructure on Amazon SageMaker AI, using Amazon EMR with Apache Spark to process billions of design elements and the custom FigmaStep framework on SageMaker Pipelines to orchestrate training. Figma trained and deployed the models behind its AI Search feature within five months, launching it at Config 2024, and ran more than 10,000 SageMaker AI training jobs in 2025 to support features including the AI prompt-to-app tool Figma Make.
The challenge
Figma needed to rapidly develop and deploy innovative AI features such as AI Search to enhance product development workflows while maintaining the reliability and performance its enterprise customers demanded, but its existing infrastructure, while well suited for traditional workloads, was not optimized for the intensive, dynamic compute and specialized hardware needs of modern AI development.
The solution
Figma built its AI infrastructure on Amazon SageMaker AI. It established secure data processing pipelines using Amazon EMR with Apache Spark to process billions of design elements from its production systems while maintaining customer privacy controls, storing the processed data in Amazon S3. Figma built AI training infrastructure on Amazon SageMaker Model Training jobs and created the custom FigmaStep framework on top of Amazon SageMaker Pipelines so engineers could compose training pipelines from reusable Python-function steps, scaling from single-machine experiments to large-scale distributed training. For deployment, Figma integrated Amazon SageMaker AI endpoints into its existing Kubernetes-based infrastructure using AWS Controllers for Kubernetes.
Reported business value
By the end of April 2024, five months after starting the project, Figma had trained and deployed the models powering AI Search, which launched at Config 2024 in June. In 2025, Figma's team ran more than 10,000 Amazon SageMaker AI training jobs, with dozens of engineers running nearly 100 workflows per month. At Config 2025, Figma doubled its platform with four new products, including Figma Make, an AI prompt-to-app tool.
Sources
Open any source and check the claim yourself — that is the point of the register.
This record was researched and written with AI assistance, and its claims were checked against the sources above. (EU AI Act art. 50 transparency notice.)
Other technology & software entries in the register.
HP crafts marketing campaigns that resonate with customers using Databricks and Uniphore
HP centralized first-party customer data on the Databricks Data + AI Platform with Delta Lake and Unity Catalog, and connected it to Uniphore's HybridCompute for federated query pushdown, cutting campaign setup from 2 weeks to 2 hours and processing 400 million records in seconds.
Building a safer and more sustainable world
Novade, a construction management software company, partnered with Databricks to modernize its analytics and ML infrastructure, moving from a homegrown Apache Airflow stack to the Databricks Data + AI Platform with Delta Lake, Unity Catalog and MLflow. The move reduced total cost of ownership by 60% and supported a 100% increase in new clients, while enabling ML models for predicting incident risk affecting worker safety and project delivery schedules.
SafeGraph optimizes geospatial data processing with Databricks and Delta Sharing
SafeGraph used the Databricks Data + AI Platform, Delta Lake, Delta Sharing and MLflow to process petabytes of geospatial data and feed predictive models, achieving 2x-10x faster spatial querying, indexing and partitioning and a 50% reduction in peak memory consumption compared to other platforms, while reducing data access time for partners from months to minutes via Delta Sharing.
Transforming Weather Forecasting with Lakeflow Jobs
AccuWeather migrated from on-premises infrastructure to Databricks and Lakeflow Jobs, working with Datadog for observability, to unify diverse weather data formats and orchestrate 4,500+ weekly jobs. Lakeflow Jobs coordinates the ingestion of multiple weather models, triggers machine learning processes that weight and blend different forecasts, and manages complex job dependencies for reinforcement training workflows used in AccuWeather's proprietary forecasting engine. AccuWeather reports 3x faster dataset development (three months to one month per dataset), a 50% reduction in unactionable alerts, and 50% cost savings on serverless job usage.
AccuWeatherLakeflow Jobs · Delta Sharing · DatadogWas this helpful?
Your feedback helps us improve our use case database