Taro Logo

Software Engineer, Machine Learning

Born on the Web, Figma helps product teams brainstorm, design and build better products through collaborative design tools.
$149,000 - $350,000
Machine Learning
Senior Software Engineer
Hybrid
1,000 - 5,000 Employees
5+ years of experience
AI · Enterprise SaaS

Description For Software Engineer, Machine Learning

Figma, a leading web-based design platform, is seeking an experienced Machine Learning Engineer to join their growing AI team. This role sits at the intersection of applied machine learning, infrastructure, and product innovation, making it an exciting opportunity for someone passionate about pushing the boundaries of AI in creative tools.

The position offers a competitive salary range of $149,000 - $350,000 for those based in San Francisco or New York, with adjustments for remote workers across the United States. The role combines technical depth in machine learning with practical product development, requiring 5+ years of industry experience with at least 3 years focused on applied ML/AI.

As a Machine Learning Engineer at Figma, you'll be responsible for designing and implementing ML models for various applications including search, discovery, ranking, and generative AI features. You'll work on building robust data pipelines, collaborating with researchers to improve model quality, and partnering with product teams to deliver impactful AI features.

The ideal candidate should have strong expertise in Python and popular ML frameworks, experience with end-to-end ML model development, and a track record of building scalable data pipelines. Knowledge of search relevance, ranking, NLP, or RAG systems is a plus, as is experience with creative or design-focused ML applications.

Figma offers a comprehensive benefits package including health, dental, and vision insurance, retirement benefits, parental leave, mental health support, and various stipends for professional development and remote work. The company culture emphasizes growth, collaboration, and making design accessible to all, making it an ideal environment for those who want to make a significant impact in the design tools space.

This role provides an opportunity to shape the future of AI in design tools while working with a talented team in a collaborative environment. Whether you're interested in building intelligent search systems, developing creative AI tools, or scaling ML infrastructure, you'll find challenging and meaningful work at Figma.

Last updated 23 days ago

Responsibilities For Software Engineer, Machine Learning

  • 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
  • Collaborate with AI researchers to iterate on datasets, evaluation metrics, and model architectures
  • Work with product engineers to define and deliver impactful AI features
  • Partner with infrastructure engineers to develop and optimize systems for training, inference, monitoring, and deployment
  • Explore new ideas and help shape the long-term AI vision at Figma

Requirements For Software Engineer, Machine Learning

Python
Go
  • 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
  • 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
  • Experience mentoring or leading others and contributing to a culture of technical excellence

Benefits For Software Engineer, Machine Learning

Medical Insurance
Dental Insurance
Vision Insurance
401k
Parental Leave
Mental Health Assistance
  • Health insurance
  • Dental insurance
  • Vision insurance
  • Retirement with company contribution
  • Parental leave
  • Mental health & wellness benefits
  • Learning & development stipend
  • Work from home stipend
  • Cell phone reimbursement
  • Annual bonus plan

Interested in this job?

Jobs Related To Figma Software Engineer, Machine Learning