over 1 year ago
Base Salary
$149k - $350k/yr
Responsibilities
- Design, build, and productionize ML models for search, discovery, ranking, RAG, and generative AI features.
- Build and maintain scalable pipelines for training and evaluation datasets, annotation systems, and human-in-the-loop workflows.
- Collaborate with AI researchers on datasets, evaluation metrics, and model architectures.
- Partner with product engineers to define and deliver AI features across Figma’s platform.
- Work with infrastructure engineers to develop and optimize training, inference, monitoring, and deployment systems.
- Explore new technical approaches and help shape Figma’s long-term AI vision.
Requirements
- At least 5 years of industry software engineering experience, including at least 3 years focused on applied machine learning or AI.
- Strong end-to-end experience developing, training, evaluating, deploying, and monitoring ML models.
- Proficiency in Python and familiarity with PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
- Experience building scalable data and annotation pipelines and evaluation systems for AI model quality.
- Experience mentoring or leading others and contributing to technical excellence and innovation.
- Preferred experience with search relevance, ranking, NLP, or RAG systems.
- Preferred experience with AI infrastructure and MLOps, including observability, CI/CD, and ML workflow automation.
- Preferred experience applying ML to creative or design-focused products.
- C++ or Go knowledge is a plus.
- Product mindset and ability to connect technical work to user outcomes and business impact.
- Strong cross-functional collaboration and communication skills.
Benefits
- Full-time role available from Figma’s US hubs or remotely within the United States.
- Health, dental, and vision benefits.
- Retirement plan with company contribution.
- Parental leave and reproductive or family planning support.
- Mental health and wellness benefits.
- Generous paid time off and company recharge days.
- Learning and development stipend.
- Work-from-home stipend and cell phone reimbursement.
- Equity, annual bonus eligibility for non-sales roles, and other benefits.
- In-person onboarding is required if hired.
Categories
About Figma
Figma builds a browser-based platform for interface design, prototyping, whiteboarding, and developer handoff used by product, design, and engineering teams. Its main products—Figma Design and FigJam—are sold as SaaS with tiered subscriptions for individuals, teams, and enterprises, plus an ecosystem of plugins and integrations. Founded in 2012 and headquartered in San Francisco, the company is privately held and enables real-time collaboration on the web and desktop apps.