Figma · 1 week ago
Software Engineer, Machine Learning
San Francisco, CA
Full-time
Hybrid
Senior Level
$149K/yr - $350K/yr
5+ years expFigma is a company on a mission to make design accessible to all. They are seeking a versatile and experienced Machine Learning / AI Engineer to join their AI team, focusing on applied machine learning, infrastructure, and product innovation.
CollaborationDeveloper ToolsProduct DesignSoftwareWeb Apps
Responsibilities
Design, build, and productionize ML 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, including annotation systems and human-in-the-loop workflows
Collaborate with AI researchers to iterate on datasets, evaluation metrics, and model architectures to improve quality and relevance
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 training, inference, monitoring, and deployment
Explore new ideas at the edge of what’s technically possible and help shape the long-term AI vision at Figma
Qualification
Required
5+ years of industry experience in software engineering, with 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 ML libraries like PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost
Experience designing and building scalable data and annotation pipelines, as well as evaluation systems for AI model quality
Experience mentoring or leading others and contributing to a culture of technical excellence and innovation
Preferred
Familiarity with search relevance, ranking, 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 ML applications
Knowledge of additional languages such as C++ or Go is a plus, but not required
A product mindset with the ability to tie technical work to user outcomes and business impact
Strong collaboration and communication skills, especially when working across functions (engineering, product, research)
Benefits
Health, dental & vision
Retirement with company contribution
Parental leave & reproductive or family planning support
Mental health & wellness benefits
Generous PTO
Company recharge days
A learning & development stipend
A work from home stipend
Cell phone reimbursement
Sales incentive pay for most sales roles
An annual bonus plan for eligible non-sales roles
Company
Figma
Figma is a collaborative design tool that enables teams to create, prototype, and test digital products on one platform.
1001-5000 employees
H1B Sponsorship
Figma has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (47)
2024 (27)
2023 (32)
2022 (35)
2021 (16)
2020 (6)
Funding
Current Stage
Public CompanyTotal Funding
$1.45BKey Investors
Durable Capital PartnersAndreessen HorowitzSequoia Capital
2025-12-08Post Ipo Equity
2025-07-31IPO
2024-07-17Secondary Market· $700M
Leadership Team
Dylan Field
CEO & Co-founder
Evan Wallace
Co-Founder
Recent News
2026-01-11
The Motley Fool
2026-01-11
2026-01-08
Company data provided by crunchbase