Software Engineer, Machine Learning - USA | Remote

About the Company

Figma is a collaborative design platform that helps teams turn ideas into products through brainstorming, prototyping, design-to-code workflows, and AI-powered tools. Its platform enables teams to streamline workflows, work together in real time, and build products from anywhere.

About the Role

Figma is seeking an experienced Machine Learning / AI Engineer to join its AI team. This role works across applied machine learning, infrastructure, and product innovation, contributing to intelligent search systems, scalable data pipelines, and AI-powered creative tools.

You will collaborate with engineers, researchers, designers, and product managers to develop ML-driven features and infrastructure that improve user productivity and create new product experiences. This is a full-time position that can be based at one of Figma's US hubs or performed remotely within the United States.

Key Responsibilities

• 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 for collecting high-quality training and evaluation datasets, including annotation systems and human-in-the-loop workflows.
• Collaborate with AI researchers to improve datasets, evaluation metrics, and model architectures.
• 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 model training, inference, monitoring, and deployment.
• Explore emerging technical possibilities and contribute to Figma's long-term AI vision.

Required Qualifications

• 5+ years of industry experience in software engineering, including at least 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 machine learning libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
• Experience designing and building scalable data and annotation pipelines, along with evaluation systems for AI model quality.
• Experience mentoring or leading others and contributing to a culture of technical excellence and innovation.

Preferred Qualifications

• Familiarity with search relevance, ranking, natural language processing (NLP), or RAG systems.
• Experience with AI infrastructure and MLOps, including observability, CI/CD, and automation for ML workflows.
• Experience working on creative or design-focused machine learning applications.
• Knowledge of additional programming languages such as C++ or Go.
• A product mindset and the ability to connect technical work with user outcomes and business impact.
• Strong collaboration and communication skills when working across engineering, product, and research teams.

Compensation & Benefits

• Annual base salary range for positions based in Figma's San Francisco or New York hub offices: $153,000–$376,000 USD.
• Compensation depends on factors including qualifications, relevant skills and experience, potential impact, role scope, market demands, and work location.
• For remote positions, the salary range is localized according to the employee's work location, at 80%–100% of the stated range. Candidates can discuss location-specific compensation with their recruiter.
• Equity for employees.
• 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.
• Annual bonus plan for eligible non-sales roles.

Work Arrangement

• Full-time position.
• Available at one of Figma's US hubs or remotely within the United States.
• Candidates are required to keep their cameras on during video interviews.
• In-person onboarding is required for successful candidates.

Equal Employment Opportunity & Accessibility

Figma is an equal opportunity workplace. The company is committed to providing equal employment opportunities regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or expression, veteran status, or any other characteristic protected by law. Qualified applicants with criminal histories are also considered in accordance with applicable legal requirements.

Figma provides reasonable accommodations for individuals with disabilities throughout the application and interview process, during essential job functions, and in accessing employment benefits and privileges. Examples include accessible interview locations, closed captioning for video calls, screen-reader-compatible written communication, and alternative interview formats.

For accommodation requests, contact accommodations-ext@figma.com.

Candidate Privacy

By applying, candidates acknowledge and agree that personal data included in their application or supporting materials will be processed in accordance with Figma's Candidate Privacy Notice.