RemoteLeaf Logo
Log in Sign up

Software Engineer, Machine Learning

4 months, 3 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 70

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

Starburst

Senior Software Engineer, Control Plane

Starburst 251-1K IT Services

Starburst is hiring a Senior Software Engineer for its globally distributed Control Plane team to unify the platform supporting Starburst Enterprise, Starburst Galaxy, and Trino across SaaS, self-managed, and air-gapped environments.

Poland Full-time Senior Software Engineer
$185k-$215k
AWS Azure GitHub Java React REST API Trino TypeScript
2 hours, 48 minutes ago
JumpCloud

Staff Software Engineer, Privileged Access Management (PAM) / Password Manager (PWM) - India

JumpCloud 251-1K Internet Software & Services

JumpCloud is seeking a remote Staff Software Engineer to architect and guide secure, scalable core-platform microservices while aligning engineering teams around long-term technical strategy and AI-enabled delivery.

India Full-time Lead Software Engineer
AWS C# CI/CD GCP Go gRPC HashiCorp Vault Kubernetes Microservices .NET Playwright PostgreSQL SQL Server TypeScript
2 hours, 48 minutes ago
JumpCloud

Senior Software Engineer - Privileged Access Management (PAM / IAM)- India

JumpCloud 251-1K Internet Software & Services

JumpCloud is hiring a remote Senior/Lead Software Engineer to evolve its core AI-powered IT management platform through secure, scalable architecture, enterprise PAM integration, reliable delivery, and AI-assisted engineering practices.

India Full-time Lead Security Engineer Software Engineer
Agile AWS C# Docker Go gRPC HashiCorp Vault Kubernetes Linux macOS Microservices .NET Playwright PostgreSQL REST API SQL Server TypeScript
3 hours, 3 minutes ago
Mozilla

Senior Software Engineer, Add-Ons

Mozilla 251-1K Internet Software & Services

Mozilla is hiring a Senior Software Engineer – Operations to modernize and operate addons.mozilla.org, building reliable systems and tools that support reviewers, developers, and millions of Firefox add-on users.

Anywhere Full-time Senior Software Engineer
Django GCP MySQL Python React TypeScript
3 hours, 18 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