AI Infrastructure Engineer for AI/ML Model Development

2 days ago


San Francisco, California, United States Figma Full time

Figma is a platform that makes design accessible to all. Born on the Web, Figma helps entire product teams brainstorm, design and build better products - from start to finish.

Job Description

We're looking for an AI engineer who has experience building data pipelines to collect high-quality data, and evaluation systems to evaluate AI models. You will be collaborating with AI researchers and product engineers closely to improve Figma's AI features, build new capabilities, and advance the state of the art in AI for creativity.

  • Build data pipelines to collect high-quality training and evaluation data for developing AI models.
  • Collaborate closely with AI product engineers to define requirements for AI features.
  • Collaborate closely with AI researchers to identify failure cases, improve datasets, evaluation metrics, and hill-climb quality of AI models.
  • Work closely with AI infra engineers to build the right infrastructure for AI model development and inference.

We'd love to hear from you if you have:

  • Expertise building infrastructure for AI/ML model development in industry.
  • Experience in software development in Python.
  • Experience working closely with AI researchers on applied AI projects to build high-quality models in a fast-paced environment.
  • Expertise improving datasets and evaluation metrics for applied ML projects.
  • Experience solving sophisticated problems and comparing alternative solutions, trade-offs, and diverse points of view to determine a path forward.

Figma believes in hiring smart, curious people who are excited to learn and develop their skills. If you're excited about this role but your past experience doesn't align perfectly with the points outlined in the job description, we encourage you to apply anyways.

What We Offer

At Figma, one of our values is Grow as you go. We believe in providing a competitive package of 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.

Salary and Benefits

The estimated annual base salary range for this role is $149,000 - $350,000 USD, depending on location and qualifications.



  • San Francisco, California, United States Naptha AI Full time

    About Naptha AIWe are seeking exceptional Software Engineering interns to join Naptha AI and contribute to building the future of AI agent infrastructure.This internship offers hands-on experience working with frontier AI technology, backed by industry veterans and technical leaders through NVIDIA Inception, Google for Startups, and Microsoft for Startups.As...


  • San Francisco, California, United States Together AI Full time

    About the RoleWe are seeking an experienced Systems Research Engineer to join our team at Together AI. As a key member of our research-driven artificial intelligence company, you will play a crucial role in researching and building the next generation AI platform.Company OverviewTogether AI is committed to creating open and transparent AI systems that drive...


  • San Francisco, California, United States Abridge AI Inc. Full time

    Abridge AI Inc. is a pioneering force in healthcare technology, utilizing artificial intelligence to empower deeper understanding and improve clinical documentation efficiency.Role OverviewWe are seeking an exceptional ML Systems Engineer to join our team, responsible for scaling and deploying machine learning models to handle increasing traffic demands and...


  • San Francisco, California, United States Abridge AI Inc. Full time

    Abridge AI Inc. is a trailblazing organization that empowers deeper understanding in healthcare through artificial intelligence.Estimated Salary: $185,000 USD - $265,000+ USD per year + EquityWe are seeking experienced Full Stack Engineers to join our growing team and help us build innovative ML-powered solutions for healthcare AI technology.About the...


  • San Francisco, California, United States Scale AI Full time

    Cloud AI Engineer Position at ScaleWe are seeking an experienced Cloud AI Engineer to join our team at Scale, a leading provider of AI solutions. As a Cloud AI Engineer, you will play a key role in designing and developing our cloud infrastructure platforms and systems.The ideal candidate will have extensive experience in software development and a deep...


  • San Francisco, California, United States Magic AI Full time

    Company OverviewMagic AI is a cutting-edge technology company dedicated to building safe Artificial General Intelligence (AGI) that accelerates humanity's progress on the world's most important problems.We believe the most promising path to safe AGI lies in automating research and code generation to improve models and solve alignment more reliably than...


  • San Francisco, California, United States Decagon AI, Inc. Full time

    We are seeking a highly skilled AI Engineer to join our team at Decagon AI, Inc. in building cutting-edge conversational AI agents for the enterprise.About the Role:This role involves evaluating and implementing advanced AI solutions in large-scale enterprise environments. The ideal candidate will have a strong background in AI/ML engineering, with 5+ years...


  • San Francisco, California, United States Together AI Full time

    Are you a skilled DevOps engineer looking to take your career to the next level? Do you have a passion for designing and building automated infrastructure pipelines? We are seeking a talented Senior DevOps Engineer to join our cloud engineering team at Together AI. About the RoleWe are hiring a highly experienced Senior DevOps Engineer to lead the...


  • San Francisco, California, United States Scale AI Full time

    About Scale AIAt Scale AI, our mission is to accelerate the development of AI applications. With 8 years of experience as the leading AI data foundry, we've helped fuel exciting advancements in AI, including generative AI, defense applications, and autonomous vehicles. Our recent Series F round has enabled us to accelerate the abundance of frontier data,...


  • San Francisco, California, United States Abridge AI Full time

    Unlock the Future of Healthcare with Abridge AIAbridge AI is a pioneering organization dedicated to revolutionizing medical conversations through AI. We are seeking an experienced Full Stack Software Engineer to join our team and help us build innovative solutions for healthcare professionals.About the RoleThis position offers a unique opportunity to design,...

  • AI Model Developer

    7 days ago


    San Francisco, California, United States Aitopics Full time

    About SupernormalSupernormal is a cutting-edge AI-powered meeting platform designed to revolutionize the way teams collaborate and communicate. Our mission is to transform spoken communication for individuals, teams, and organizations by automating meeting notes, agendas, and insights.Job SummaryWe are seeking a highly skilled Machine Learning Engineer to...


  • San Francisco, California, United States Abridge AI Inc. Full time

    Abridge AI Inc. is a pioneering organization in the field of healthcare technology, dedicated to harnessing the power of artificial intelligence (AI) to revolutionize medical conversations and improve clinical documentation efficiency.Estimated Salary: $190,000 - $270,000 per yearThis role requires an exceptional Full Stack Engineer to join our team, working...

  • AI Engineer

    5 days ago


    San Francisco, California, United States Databricks Inc. Full time

    About the RoleWe are seeking an experienced AI Engineer to join our team in developing large language models. As a key member of our platform engineering group, you will be responsible for designing and implementing the infrastructure that powers our generative AI platform.Your primary focus will be on building out our ML model development lifecycle,...


  • San Francisco, California, United States Abridge AI Inc. Full time

    Unlock the Potential of Healthcare with AbridgeAbridge AI Inc. is revolutionizing the healthcare industry with cutting-edge AI technology, empowering clinicians to focus on patient care while streamlining clinical documentation processes.About the RoleWe are seeking an experienced Transformative AI Systems Architect to join our team and play a pivotal role...


  • San Francisco, California, United States Scale AI Full time

    Research Role OverviewScale AI's Generative AI team is pushing the boundaries of artificial intelligence by developing innovative models, algorithms, and supervision techniques. As a Senior AI Research Scientist for Generative Models, you will play a critical role in advancing our research agenda and driving product development. Your expertise in Generative...


  • San Francisco, California, United States Naptha AI Full time

    Naptha AI is seeking a highly skilled AI Research Scientist to join our team. This role focuses on advancing the state of the art in multi-agent systems and agent interoperability.About the RoleWe are looking for an exceptional researcher who can design and implement novel architectures for scalable multi-agent systems, develop frameworks for agent...


  • San Francisco, California, United States Scale AI Full time

    OverviewSkyrocket the advancement of AI across industries at Scale, a pioneering company in AI research and development. Our mission is to accelerate the transition from traditional software to AI, empowering organizations to build and deploy cutting-edge models.

  • Software Engineer

    1 week ago


    San Francisco, California, United States Stack AI Full time

    About Stack AIWe're a fast-growing startup on a mission to democratize access to Large Language Models. Our user-friendly and intuitive No-Code platform integrates the best AI models, common data sources, and SaaS tools.Our Traction is impressive: launched 8 months ago with over 65,000 users and 300+ paying customers, including public companies and...


  • San Francisco, California, United States GEICO Full time

    Job SummaryGEICO is seeking an experienced Senior Machine Learning Engineer to drive the development and operation of high-performance, low-maintenance AI/ML platforms and applications. As a key member of the Document Platform Engineering organization, you will play a critical role in shaping the technical roadmap of the document intelligence sub-area,...


  • San Francisco, California, United States Truva AI Full time

    Transformative Opportunity at Truva AIWe are a pioneering force in SaaS innovation, specializing in automating tasks and optimizing workflows with cutting-edge Large Language Models (LLMs). Our mission is to deliver unparalleled operational efficiency and drive technological advancements.Backed by top VCs such as YCombinator and Fintech Collective, we are...