Taro Logo

Software Engineer, Machine Learning

Figma is a design platform that helps teams create, prototype, and collaborate on digital products in real-time.
$149,000 - $350,000
Machine Learning
Senior Software Engineer
Remote
1,000 - 5,000 Employees
5+ years of experience
AI · Enterprise SaaS

Job Description

Figma is seeking an experienced Machine Learning / AI Engineer to join their growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. This role offers a unique opportunity to shape the future of design and collaboration through AI-powered features.

The position involves building intelligent search systems, crafting scalable data pipelines, and enhancing AI-powered creativity tools that will drive user productivity and shape new product experiences. As a key member of the AI team, you'll work on both foundational systems and user-facing capabilities, collaborating across multiple teams including engineers, researchers, designers, and product managers.

The ideal candidate brings 5+ years of industry experience in software engineering, with at least 3 years focused on applied machine learning or AI. Strong expertise in Python and ML frameworks is essential, along with experience in end-to-end ML model development and scalable data pipeline construction. The role offers competitive compensation ranging from $149,000 to $350,000 USD for hub locations, with comprehensive benefits including health insurance, retirement contributions, and various stipends for professional development and remote work.

This is a high-impact opportunity to work at a company that's revolutionizing design tools and collaboration. You'll be part of a team that values technical excellence, innovation, and cross-functional collaboration, while helping to advance the state of AI at Figma. The position can be held from US hubs or remotely within the United States, offering flexibility in work location while maintaining strong connection to the team and mission.

Last updated 15 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 for 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 AI features
  • Partner with infrastructure engineers to develop and optimize systems
  • Explore new ideas and help shape long-term AI vision

Requirements For Software Engineer, Machine Learning

Python
  • 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
  • Experience designing and building scalable data and annotation pipelines
  • Experience mentoring or leading others

Benefits For Software Engineer, Machine Learning

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

Related Jobs