Senior ML Engineer — AI & Design Infrastructure

Nubeero Limited; San Francisco, Northern (CA, KY); Hybrid; USD 153,000 - 376,000; Full time

Job summary

Figma is seeking a versatile Machine Learning / AI Engineer to join our AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. You’ll collaborate with engineers, researchers, designers, and product managers to deliver ML-driven features across the platform.

You have 5+ years in software engineering with 3+ years in ML, strong Python skills, and experience building end-to-end models, data pipelines, and evaluation systems.

Qualifications

• 5+ years of industry 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.
• Experience mentoring or leading others and contributing to a culture of technical excellence and innovation.

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.
• Work with product engineers to define and deliver impactful AI features across Figma’s platform.
• Partner with infrastructure engineers to develop and optimize systems for training, inference, monitoring, and deployment.
• Explore new ideas at the edge of what’s technically possible and help shape the long-term AI vision at Figma.

Skills

5+ years in software engineering with

Python & ML libraries (PyTorch, Tensor

end-to-end ML lifecycle

ML data pipelines & annotation systems

NLP / RAG familiarity

Tools

PyTorch

TensorFlow

Scikit-learn

Spark MLlib

XGBoost

Job description

Figma is seeking a versatile Machine Learning / AI Engineer to join our AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. You’ll collaborate with engineers, researchers, designers, and product managers to deliver ML-driven features across the platform.

You have 5+ years in software engineering with 3+ years in ML, strong Python skills, and experience building end-to-end models, data pipelines, and evaluation systems.