AI Applied Scientist

Job summary

Responsibilities

Figma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma’s platform helps teams bring ideas to life—whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI.

• You will be driving fundamental and applied research in AI. You will explore the boundaries of what is possible with the current technology set to build best in class models for…
• You will be combining industry best practices and a first-principles approach to build cutting edge Generative AI models, using techniques like Supervised Finetuning (SFT)…
• Work in concert with product and infrastructure engineers to improve Figma’s products through AI powered features
• Collaborate closely with product managers and engineers to transform user feedback into requirements for AI systems
• Build evaluation systems to measure and improve quality of AI features in Figma products

What they're looking for

• Extensive experience in building generative AI features through prompt engineering, and fine tuning models in production environments
• Experience working on deep learning and generative AI frameworks like PyTorch, JAX, HuggingFace etc
• Experience training LLMs with Reinforcement Learning techniques such as preference-based RL (DPO, PPO) and/or RL with verifiable rewards (RLVR) such as GRPO/DAPO.
• 4+ years in Generative AI, and 6+ years of experience in one or more of the following areas: machine learning, natural language processing/understanding, computer vision
• Strong software engineering skills with 5+ years of experience in programming languages (Python, C++, Java or R)

Tech stack

• PyTorch
• JAX
• Python
• C++
• Java
• Figma