RemoteLeaf Logo
Log in Sign up

Software Engineer, Machine Learning

5 months, 2 weeks ago
United States
Full-time
Senior
Software Engineer
Software Development
C++ CI/CD Generative AI Go Machine Learning MLOps NLP Python PyTorch Scikit-learn TensorFlow XGBoost
Apply Now
Figma

Figma

Figma provides a collaborative web application for interface design that enables users to seamlessly design, prototype, and gather feedback in real time, while also offering additional offline capabilities through desktop applications for macOS and Win...

Internet Software & Services
1K-5K
Founded 2012
$333M raised
View All Jobs 64

Description

  • Design, build, and productionize machine learning models for search, discovery, ranking, retrieval-augmented generation (RAG), and generative AI features.
  • Build and maintain scalable data pipelines to collect training and evaluation datasets, including annotation systems and human-in-the-loop workflows.
  • Collaborate with AI researchers to refine datasets, evaluation metrics, and model architectures.
  • Work with product engineers to define and deliver AI features across Figma’s platform.
  • Partner with infrastructure engineers to develop and optimize training, inference, monitoring, and deployment systems.
  • Explore new technical ideas and help shape Figma’s long-term AI vision.
  • Collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver ML-driven features and infrastructure.

Requirements

  • 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.
  • Familiarity with ML libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
  • Experience designing and building scalable data and annotation pipelines.
  • Experience building evaluation systems for AI model quality.
  • Experience mentoring or leading others and contributing to technical excellence and innovation.
  • Familiarity with search relevance, ranking, NLP, or RAG systems is preferred.
  • Experience with AI infrastructure and MLOps, including observability, CI/CD, and automation for ML workflows, is preferred.
  • Experience working on creative or design-focused ML applications is preferred.
  • Knowledge of additional languages such as C++ or Go is a plus.
  • A product mindset with the ability to connect technical work to user outcomes and business impact is preferred.
  • Strong collaboration and communication skills, especially across engineering, product, and research teams is preferred.

Benefits

  • Annual base salary range of $153,000 to $376,000 USD, depending on location and level.
  • Equity as part of the compensation package.
  • Health, dental, and vision insurance.
  • Retirement plan with company contribution.
  • Parental leave and reproductive or family planning support.
  • Mental health and wellness benefits.
  • Generous PTO and company recharge days.
  • Learning and development stipend, work-from-home stipend, and cell phone reimbursement.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

MongoDB

Senior Software Engineer, Query Execution

MongoDB 1K-5K Internet Software & Services

MongoDB is hiring a Software Engineer for its Query Execution Team in Denmark to build and improve high-performance, reliable, and modular distributed query systems powering critical customer workloads.

Denmark Full-time Senior Software Engineer
C C++ MongoDB Rust
8 hours, 14 minutes ago
MongoDB

Senior Software Engineer, Query Execution

MongoDB 1K-5K Internet Software & Services

MongoDB is hiring a software engineer in Germany to develop and improve reliable, high-performance distributed query execution systems that support customers’ critical workloads.

Germany Full-time Senior Software Engineer
AWS Azure C C++ GCP MongoDB Rust
8 hours, 14 minutes ago
MongoDB

Senior Software Engineer, Query Execution

MongoDB 1K-5K Internet Software & Services

MongoDB is hiring a Software Engineer for its Query Execution Team in Spain to build and improve high-performance, reliable, and modular distributed query systems that support customers’ critical workloads.

Spain Full-time Senior Software Engineer
AWS Azure C C++ GCP MongoDB Rust
8 hours, 29 minutes ago
MongoDB

Senior Software Engineer, Query Execution

MongoDB 1K-5K Internet Software & Services

MongoDB is hiring a Software Engineer for its Query Execution Team to develop and improve high-performance, reliable, modular distributed query systems powering critical customer workloads; the role can be based remotely in Bulgaria.

Bulgaria Full-time Senior Software Engineer
C C++ MongoDB Rust
8 hours, 29 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers