Software Engineer - Machine Learning

Figma is seeking a Machine Learning/AI Engineer to work on applied ML, infrastructure, and product innovation. The role involves designing and productionizing ML models for Search, Discovery, Ranking, RAG, and generative AI, building scalable data pipelines, collaborating with researchers and product engineers, and shaping the long-term AI vision. This is a full-time role based in US hubs or remote.

5+ years of software engineering experience with 3+ years in applied ML/AI. Strong experience in end-to-end ML model development, proficiency in Python and ML libraries (PyTorch, TensorFlow, etc.), experience with scalable data pipelines and evaluation systems, and mentoring/leading others. Preferred: familiarity with search relevance, ranking, NLP, RAG, MLOps, and creative ML applications.