Software Engineer, Machine Learning (WFH)

About the Company

Figma is a collaborative design platform that helps teams turn ideas into products. From brainstorming and prototyping to translating designs into code and iterating with AI, Figma enables teams to streamline workflows, accelerate product development, and collaborate in real time from anywhere in the world.

The company is committed to making design accessible to everyone and bringing together passionate creatives and builders to shape the future of design and collaboration.

About the Role

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join its growing AI team. This role sits at the intersection of applied machine learning, infrastructure, and product innovation, with opportunities to develop intelligent search systems, scalable data pipelines, and AI-powered creative tools.

You will collaborate with engineers, AI researchers, designers, and product managers across multiple teams to deliver high-quality machine learning features and infrastructure. Your work will contribute to foundational AI systems, improve user productivity, and shape new product experiences across the Figma platform.

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 to collect high-quality training and evaluation datasets.
• Develop annotation systems and human-in-the-loop workflows to improve data quality and model performance.
• Collaborate with AI researchers to refine datasets, evaluation metrics, and model architectures to improve model quality and relevance.
• Partner with product engineers to define, develop, and deliver impactful AI features across Figma’s platform.
• Work with infrastructure engineers to develop and optimize systems for model training, inference, monitoring, and deployment.
• Explore emerging technical approaches and contribute to Figma’s long-term AI vision.
• Contribute to technical excellence, innovation, and knowledge sharing across the engineering organization.

Required Qualifications

• At least five years of industry experience in software engineering, including at least three years focused on applied machine learning or AI.
• Strong experience with end-to-end machine learning model development, including training, evaluation, deployment, and monitoring.
• Proficiency in Python.
• Familiarity with machine learning libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
• Experience designing and building scalable data pipelines, annotation workflows, and 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, continuous integration and continuous delivery (CI/CD), and automation for machine learning workflows.
• Experience developing machine learning applications for creative or design-focused products.
• Knowledge of additional programming languages, such as C++ or Go.
• A product-oriented mindset and the ability to connect technical work with user outcomes and business impact.
• Strong collaboration and communication skills, particularly when working across engineering, product, and research teams.

Employment Type and Work Arrangement

• Full-time position.
• Available remotely within the United States or from one of Figma’s US hub offices.
• Candidates must keep their cameras on during video interviews as part of Figma’s hiring process.
• In-person onboarding is required for successful candidates.

Compensation and Benefits

• Annual base salary range of $153,000 to $376,000 for the role, with the stated range applying to positions based in Figma’s San Francisco or New York hub offices.
• Job level and actual compensation will depend on factors including qualifications, relevant skills and experience, potential impact, role scope, market conditions, and work location.
• For remote positions, the salary range is adjusted according to employee location, using a factor between 80% and 100% of the stated range. Candidates should discuss location-specific compensation with their recruiter.
• Equity for employees.
• Health, dental, and vision benefits.
• Retirement benefits with company contributions.
• 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 and sales incentive pay for most sales roles.

Compensation and benefits are subject to change.

Equal Employment Opportunity

Figma is an equal opportunity workplace committed to creating an inclusive team with diverse perspectives, experiences, and opinions. The company provides 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 are also considered regardless of criminal history, consistent with legal requirements.

Figma encourages candidates to apply even if their experience does not align perfectly with every listed qualification, as they may be a strong fit for this or another role.

Accessibility and Reasonable Accommodations

Figma provides reasonable accommodations for individuals with disabilities throughout the application process, interviews, essential job functions, and access to employment benefits and privileges.

Accommodations may include accessible interview locations, closed captioning for video calls, screen-reader-compatible written communication, or adjustments to the format or mode of interviews.

Candidates who require accommodations can contact [email protected].

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.