✦ Luna Orbit — AI & Machine Learning

Data Platform Engineer

at Figma

📍 San Francisco, CA • New York, NY • United States Onsite Posted April 16, 2026
Type Full-Time
Experience mid
Exp. Years 5+ years
Education Not specified
Category AI & Machine Learning

Figma is hiring a Data Platform Engineer to build the foundational ML and data platform that powers AI-driven products and self-serve analytics. The role focuses on owning the AI data agent layer, implementing prompt-processing pipelines and usage analytics, and operationalizing model serving and data workflows in production.

  • Lead Figma’s AI data agent workstreams (data-agent layer, prompt-processing pipelines, instrumenting interactions, prompt-based usage analytics)
  • Own and evolve Figma’s ML and data platform (model serving, feature pipelines, workflow orchestration, CI/CD for models, production monitoring)
  • Build product-facing data systems and data products to make models and data first-class components
  • Ship platform tooling for Data Science model deployment, iteration, and operation (feature stores, rollout systems, observability)
  • Design and scale infrastructure for AI-assisted and natural language interfaces to data

You will design and evolve the ML and data platform including model serving, feature pipelines, workflow orchestration, CI/CD for models, and production monitoring. You will also build product-facing data systems, platform tooling for ML deployment/iteration, and infrastructure for AI-assisted and natural language interfaces to enable self-serve analytics.

The ideal candidate is a mid-level data platform/ML engineer with 5+ years building and operating data platform and AI/ML systems in production. They are strong in Python and have hands-on experience with model serving, feature pipelines, workflow orchestration, and CI/CD for models, plus observability/monitoring for production reliability.

5+ years of experience in data platforminfrastructureor machine learning engineeringat least 1+ years working on AI or ML systemsExperience building and operating end-to-end ML systems in production (trainingevaluationdeploymentmonitoring)Strong programming skills in PythonExperience designing ML infrastructure (model servingfeature pipelinesworkflow orchestration)Proven ability to work cross-functionally and drive projects across Data ScienceEngineeringInfrastructureand Product
building prompt-processing pipelines for AI data agentinstrumenting interactions for prompt-based usage analytics
data platforminfrastructure engineeringmachine learning engineeringAI or ML systemsend-to-end ML systems in productionPythonmodel servingfeature pipelinesworkflow orchestrationCI/CD for modelsproduction monitoringprompt-processing pipelinesprompt-based usage analyticsfeature storesrollout systemsobservabilityself-serve analyticsdata contractsSLAssystem design
data platforminfrastructure engineeringmachine learning engineeringAI or ML systemsend-to-end ML systems in production (trainingevaluationdeploymentmonitoring)Pythonmodel servingfeature pipelinesworkflow orchestrationCI/CD for modelsproduction monitoringdata contractsSLAssystem designfeature storesrollout systemsobservabilityprompt-processing pipelinesprompt-based usage analyticsinstrumenting interactionsself-serve analyticstrainingevaluationdeploymentmonitoringscalable system architecturesdeveloper experienceplatform toolingtooling abstractionsmachine learning and data platform
cross-functional collaborationdriving projectscross-functional platform initiativescommunicating and aligning stakeholdersproblem-solvingproject leadershipengineering discipline for reliable scalable systems
Industry SaaS
Job Function Build and operate Figma’s ML and data platform and AI data-agent systems to enable self-serve analytics and production-grade AI.
Role Subtype MLOps Engineer
Tech Domains Python, AI & Machine Learning, Linux, Kubernetes, Cloud & Infrastructure, IT Infrastructure Admin
Data Platform EngineerData EngineeringAI data agentself-serve analyticsprompt-processing pipelinesprompt-based usage analyticsML and data platformmodel servingfeature pipelinesworkflow orchestrationCI/CD for modelsproduction monitoringfeature storesobservabilityinfrastructuremachine learning engineeringend-to-end ML systemstrainingevaluationdeploymentmonitoringPythondata contractsSLAssystem designscalable system architectures

Must have 5+ years of experience in data platform, infrastructure, or machine learning engineering, Must have at least 1+ years working on AI or ML systems, Must have experience building and operating end-to-end ML systems in production (training, evaluation, deployment, monitoring), Must have strong programming skills in Python

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile