Figma is seeking a Software Engineer, Machine Learning to join their team in San Francisco, CA. The role involves designing, building, and productionizing machine learning models and infrastructure to enhance AI-powered features across Figma's platform.
About the Role
As a Software Engineer, Machine Learning, you will collaborate with engineers, researchers, designers, and product managers to deliver high-quality ML-driven features. Your responsibilities will include building scalable data pipelines, enhancing AI tools, and exploring innovative ideas to shape Figma's AI vision.
About You
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.
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.
Strong collaboration and communication skills.
Benefits
Equity options for employees.
Comprehensive health, dental, and vision insurance.
Retirement plan with company contribution.
Parental leave and reproductive or family planning support.
Mental health and wellness benefits.
Generous paid time off (PTO) and company recharge days.
Learning and development stipend.
Work from home stipend and cell phone reimbursement.