Software Engineer, Machine Learning

Figma Logo

Figma

πŸ’΅ $149k-$350k
πŸ“Remote - United States

Summary

Join Figma's growing AI team as a Machine Learning/AI Engineer, contributing to the development of intelligent search systems, scalable data pipelines, and AI-powered creativity tools. Collaborate with engineers, researchers, designers, and product managers to deliver high-quality ML-driven features and infrastructure. This high-impact role involves designing, building, and productionizing ML models, building and maintaining data pipelines, collaborating with researchers to improve model quality, working with product engineers to deliver AI features, and partnering with infrastructure engineers to optimize systems. You will explore new ideas and shape Figma's long-term AI vision. This full-time role can be based in a US hub or remotely within the United States. Figma offers a competitive compensation and benefits package.

Requirements

  • 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

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

Preferred Qualifications

  • 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

Share this job:

Disclaimer: Please check that the job is real before you apply. Applying might take you to another website that we don't own. Please be aware that any actions taken during the application process are solely your responsibility, and we bear no responsibility for any outcomes.