Machine Learning Engineer • Mid-Level

Who you are

You have a strong background in machine learning and AI, with experience designing and building ML models that drive user productivity and enhance product experiences. Your expertise in Python allows you to create scalable data pipelines and work with high-quality datasets for training and evaluation. You thrive in collaborative environments, working closely with engineers, researchers, and product managers to deliver impactful ML-driven features. You are passionate about advancing the state of AI and enjoy tackling complex challenges at the intersection of technology and creativity.

You have a proven track record of productionizing machine learning models, ensuring they are robust and scalable. Your experience includes working with generative AI features and intelligent search systems, which are critical to enhancing user experiences at Figma. You understand the importance of data quality and are skilled in building and maintaining data pipelines that support machine learning initiatives. You are comfortable navigating the complexities of AI infrastructure and are eager to contribute to innovative product solutions.

What you'll do

At Figma, you will design, build, and productionize machine learning models for various applications, including search, discovery, and generative AI features. You will collaborate with cross-functional teams to identify opportunities for ML integration and drive the development of high-quality features that enhance user workflows. Your role will involve building and maintaining scalable data pipelines to collect and process training datasets, ensuring that the models you develop are well-informed and effective.

You will work closely with product managers and designers to understand user needs and translate them into technical requirements for ML features. Your contributions will directly impact the user experience, helping teams streamline their design processes and collaborate more effectively. You will also be responsible for monitoring the performance of deployed models and iterating on them based on user feedback and performance metrics. This is a high-impact role where your work will shape the future of design and collaboration at Figma.

What we offer

Figma provides a dynamic work environment where creativity and innovation are encouraged. You will have the opportunity to work on cutting-edge AI technologies and contribute to a mission-driven company that values design accessibility. We offer competitive compensation and benefits, along with a supportive culture that fosters professional growth and collaboration. Join us in shaping the future of design and collaboration through the power of machine learning.