Software Engineer, Machine Learning - USA | Remote

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

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

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