Position Details

Type Full-Time

Experience mid

Exp. Years 5+ years

Education Not specified

Category AI & Machine Learning

About this role

Figma seeks a seasoned Machine Learning Engineer to develop AI-driven features, scalable data pipelines, and advanced AI infrastructure to enhance user productivity and product innovation.

Key Responsibilities

Design and productionize ML models
Build scalable data pipelines
Collaborate with researchers and engineers
Optimize AI systems
Drive AI product innovation

Technical Overview

Environment includes Python, PyTorch, TensorFlow, Scikit-learn, Spark MLlib, XGBoost, with a focus on AI models, data pipelines, and MLOps.

Ideal Candidate

The ideal candidate is an experienced AI/ML engineer with over 5 years in software engineering, specializing in applied machine learning, data pipelines, and AI infrastructure, with strong collaboration skills.

Must-Have Skills

5+ years of industry experience in software engineering3+ years focused on applied machine learning or AIProficiency in PythonExperience with ML libraries like PyTorchTensorFlowDesigning scalable data pipelinesExperience with AI infrastructure and MLOps

Nice-to-Have Skills

Search relevanceRankingNLPRAG systemsAI infrastructureCI/CD for ML workflowsCreative AI tools

Tools & Platforms

PythonPyTorchTensorFlowScikit-learnSpark MLlibXGBoost

Required Skills

PythonPyTorchTensorFlowScikit-learnSpark MLlibXGBoostML modelsData pipelinesAI infrastructureMLOps

Hard Skills

PythonPyTorchTensorFlowScikit-learnSpark MLlibXGBoostML modelsData pipelinesAI infrastructureMLOps

Soft Skills

collaborationproblem-solvingcommunicationmentoringinnovation

Certifications

Preferred

None specified

Industry & Role

Industry SaaS

Job Function Developing and deploying AI and machine learning systems for product enhancement

Role Subtype Machine Learning Engineer

Tech Domains Python, PyTorch, TensorFlow, Scikit-learn, Spark MLlib

Keywords for Your Resume

machine learning engineerAI engineerML modelsdata pipelinesAI infrastructureMLOpsPythonPyTorchTensorFlowscikit-learnspark mllibxgboostsearch relevancerankingnlprag systemsgenerative AIAI toolsscalable systemspythonpytorchtensorflow

Deal Breakers

Less than 5 years of experience, No experience with ML libraries like PyTorch or TensorFlow, Lack of experience in scalable data pipelines, Inability to work onsite in San Francisco

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile

Software Engineer, Machine Learning

Get matched to jobs like this