Software Engineer, Machine Learning (WFH)

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., * 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.

• 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., * 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.

• 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.

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.