data
Posted Apr 15Software Engineer, Machine Learning
at Figma
San Francisco, United StatesOn-site
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.
Requirements
- Figma’s platform helps teams bring ideas to life—whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI.
- From idea to product, Figma empowers teams to streamline workflows, move faster, and work together in real time from anywhere in the world.
- 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.
- Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma.
- You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure.
- Work with product engineers to define and deliver impactful AI features across Figma’s platform.
- Explore new ideas at the edge of what’s technically possible and help shape the long-term AI vision at Figma. We’d love to hear from you If you have:
- 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. •
- 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.
- At Figma, one of our values is Grow as you go.
- Additionally, if hired you will be required to attend in person onboarding.
Experience
- 5+ years of industry
Benefits
- If based in Figma’s San Francisco or New York hub offices, this role has the annual base salary range stated below.
- Job level and actual compensation will be decided based on factors including, but not limited to, individual
- For roles that are available to be filled remotely, the pay range is localized according to employee work location by a factor of between 80% and 100% of range.
- Figma offers equity to employees, as well a 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. Figma also offers sales incentive pay for most sales roles and an annual bonus plan for eligible non-sales
- Figma’s compensation and
- Annual Base Salary Range: $153,000 — $376,000 USD
Contact
- If you require accommodation, please reach out to accommodations-ext@figma.com .
Additional details
- Figma is growing our team of passionate creatives and builders on a mission to make design accessible to all.
- If you're excited to shape the future of design and collaboration, join us!
- This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities.
- This is a full time role that can be held from one of our US hubs or remotely in the United States.
- Partner with infrastructure engineers to develop and optimize systems for training, inference, monitoring, and deployment.
- Strong collaboration and communication skills, especially when working across functions (engineering, product, research).
- We believe in hiring smart, curious people who are excited to learn and develop their skills.
- experience doesn’t align perfectly with the points outlined in the job description, we encourage you to apply anyways. You may be just the right candidate for this or other roles.
- qualifications objectively assessed during the interview process (including skills and prior relevant experience, potential impact, and scope of role), market demands, and specific work location.
- The listed range is a guideline, and the range for this role may be modified.