RemoteLeaf Logo
Log in Sign up

Staff Machine Learning Engineer

13 hours, 54 minutes ago
United States
Full-time
Senior
Machine Learning Engineer
Software Development
Apache Airflow Apache Spark Flink Machine Learning Python PyTorch
Apply Now
Unity

Unity

Unity is the top platform for real-time 3D content creation, empowering creators across industries to bring their ideas to life with interactive 2D and 3D content.

Internet Software & Services
5K-10K
Founded 2004
View All Jobs 52

Description

  • Design and operate large-scale data pipelines that generate training datasets for machine learning training and experimentation.
  • Develop infrastructure that supports distributed training workflows using tools such as PyTorch, Ray Data, and Ray Train.
  • Integrate ML pipelines with workflow orchestration systems such as Flyte, Airflow, or similar tools to support reliable multi-stage workflows.
  • Improve reproducibility and observability through dataset validation, monitoring, and automated testing.
  • Optimize performance and resource utilization across distributed compute systems used for data processing and model training.
  • Partner closely with ML engineers to support large-scale experimentation and model iteration.
  • Lead architectural improvements to keep offline ML pipelines scalable, reliable, and cost-efficient.
  • Work across ML engineering and platform teams to ensure pipelines can handle growing data volumes and training complexity.

Requirements

  • Strong experience building large-scale machine learning pipelines.
  • Experience with distributed computing frameworks such as Ray, Spark, or Flink.
  • Familiarity with the Ray ecosystem, including Ray Data and Ray Train, for distributed data processing and model training.
  • Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines.
  • Deep experience designing and operating production-grade data pipelines.
  • Strong programming skills in Python and experience with large-scale distributed workloads.
  • Experience with modern data infrastructure, including data lakes, warehouses, orchestration systems, and streaming platforms.
  • Strong systems thinking with the ability to reason about performance, scalability, reliability, and cost tradeoffs in distributed systems.
  • Proven ability to lead technical direction and influence architectural decisions across teams without formal authority.
  • Strong English communication skills for frequent professional verbal and written collaboration with global teams.
  • Relocation support is not available for this position.
  • Work visa or immigration sponsorship is not available for this position.

Benefits

  • Base salary range of $178,300 to $267,500 USD.
  • Comprehensive health, life, and disability insurance.
  • Employee stock ownership.
  • Competitive retirement or pension plans.
  • Generous vacation and personal days.
  • Support for new parents through leave and family-care programs.
  • Mental health and wellbeing programs and support.
  • Commute subsidy.
  • Training and development programs.
  • Volunteering and donation matching program.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Cribl

Staff Machine Learning Engineer, AI Researcher

Cribl 251-1K IT Services

Cribl is hiring a remote-first machine learning engineer to help build AI-enabled security and observability products that solve real customer problems.

United States Full-time Senior Machine Learning Engineer
$230k-$275k
Computer Vision Feature Engineering Kubeflow Machine Learning MLflow MLOps NLP Python PyTorch Reinforcement Learning TensorFlow
11 hours, 9 minutes ago
Coinbase

Staff Machine Learning Engineer - Platform (Core AI Automation)

Coinbase 1K-5K Capital Markets

Coinbase is hiring a Machine Learning Engineer for its Core Automation Team to build AI infrastructure and automation that improve customer support, compliance operations, and AI-powered customer interactions on its onchain platform.

United States Full-time Lead Machine Learning Engineer
$218k-$256k
Apache Airflow Apache Spark Blockchain Computer Vision Databricks Deep Learning Flink Generative AI Kafka LLM Machine Learning NLP Python Snowflake
11 hours, 9 minutes ago
Veriff

Software Engineer - ML Platform

Veriff 51-250 IT Services

Veriff’s ML Platform team is hiring a software or ML engineer to build the systems that support machine learning development, experimentation, observability, and scalable model deployment.

Spain Full-time Mid Level Machine Learning Engineer Software Engineer
Apache Spark dbt Grafana Kubeflow MLflow MLOps Prometheus Python Snowflake SQL
11 hours, 9 minutes ago
Samsara

Staff ML Engineer - ML Infrastructure

Samsara 1K-5K IT Services

Samsara is hiring a Staff / Senior Staff Machine Learning Infrastructure Engineer in Canada to lead the end-to-end ML platform for Safety AI and adjacent product areas that improve real-world operational safety.

Canada Full-time Lead Machine Learning Engineer
$196k-$270k
Apache Spark AWS Computer Vision Embedded Systems IoT Kubernetes LLM Machine Learning
11 hours, 39 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