RemoteLeaf Logo
Log in Sign up

Software Engineer, Machine Learning

1 hour, 39 minutes 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 58

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

Natera

LIMS Systems Engineer (Contract)

Natera 1K-5K Pharmaceuticals

Natera is hiring a LIMS Systems Engineer for a one-year contract to support LabKey LIMS configuration, implementation, and optimization for RUO laboratory operations and related systems.

United States Contract Junior Database Administrator Software Engineer
$0k-$0k
HIPAA JSON Python REST API SQL XML
4 minutes ago
Tines

Software Engineer (full-stack)

Tines 51-250 Construction & Engineering

Software engineer at Tines to help build and improve authentication, identity, access control, audit logging, and credential-related workflows on a secure, customer-facing platform.

United States Full-time Mid Level Software Engineer
$165k-$181k
AWS HTTP PostgreSQL React Redis Ruby Ruby on Rails TypeScript
4 minutes ago
Versaterm

Software Engineer II - DEMS

Versaterm 51-250 Internet Software & Services

Versaterm is hiring a Software Engineer II to help build a new standalone Digital Evidence Management System and support the foundation of secure public safety software across the company’s product ecosystem.

Canada Full-time Mid Level Software Engineer
Angular AWS Celery Docker Flask Git GitHub Actions Jest Kanban Next.js Node.js OpenAPI Playwright PostgreSQL Pytest Python React Redis Scrum SQLAlchemy Tailwind CSS Terraform TypeScript
4 minutes ago
Elastic

Principal Software Engineer II - Observability

Elastic 1K-5K Internet Software & Services

Elastic is seeking a Principal Software Engineer to serve as a Tech Lead on the Observability Experience Team, shaping end-to-end experiences for logs, metrics, and traces across the company’s cloud-based Search AI platform.

Greece Full-time Lead Site Reliability Engineer (SRE) Software Engineer
19 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