Software Engineer, Machine Learning

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation.

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.

Benefits

• Equity
• Competitive package of additional benefits, including 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, and cell phone reimbursement.
• Sales incentive pay for most sales roles and an annual bonus plan for eligible non-sales roles.