RemoteLeaf Logo
Log in Sign up

Senior/Staff Machine Learning Engineer, Data Infrastructure

1 month ago
China
Senior
Infrastructure Engineer
DevOps and Infrastructure
Apache Airflow Apache Spark Machine Learning Python
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 125

Description

  • Develop infrastructure for batch and streaming big data processing using technologies such as Flink, Spark, and Ray.
  • Design and operate large-scale data pipelines that generate training datasets for machine learning training and experimentation.
  • Integrate data pipelines with workflow orchestration systems such as Flyte, Airflow, or similar tools.
  • Improve reproducibility and observability through dataset validation, monitoring, and automated testing.
  • Optimize performance and resource utilization across distributed compute systems used for data processing.
  • Partner closely with ML engineers to support efficient large-scale experimentation and model iteration.
  • Lead architectural improvements to keep offline data pipelines scalable, reliable, and cost-efficient.

Requirements

  • Experience with distributed computing frameworks such as Flink, Spark, or Ray.
  • Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines.
  • Experience optimizing big data pipelines and infrastructure for cost efficiency.
  • Strong programming skills in Python.
  • Experience working with large-scale distributed workloads.
  • Experience with modern data infrastructure, including data lakes, warehouses, orchestration systems, and streaming platforms.
  • Strong systems thinking and 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 verbal and written collaboration with global colleagues.

Benefits

  • Comprehensive health, life, and disability insurance.
  • Commute subsidy.
  • Employee stock ownership.
  • Competitive retirement and pension plans.
  • Generous vacation and personal days.
  • Support for new parents through leave and family-care programs.
  • Mental health and wellbeing programs and support.
  • Training and development programs.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Cellanome

Senior Director, Artificial Intelligence

Cellanome 1-10 Biotechnology

Cellanome is hiring a Senior Director of Artificial Intelligence to lead AI/ML strategy and hands-on model development for its computational biology platform.

United States Full-time Executive AI Engineer Machine Learning Engineer
$288k-$330k
Deep Learning Machine Learning Python PyTorch TensorFlow
18 hours, 58 minutes ago
Reddit

Senior Machine Learning Engineer, ML Efficiency

Reddit 1K-5K Internet Software & Services

Reddit is hiring a senior engineer to build an Ads ML Efficiency function that makes model training and inference faster, cheaper, safer, and more scalable.

United States Full-time Senior Machine Learning Engineer
$217k-$303k
Machine Learning PyTorch
18 hours, 58 minutes ago
CloudLinux

Junior Infrastructure Engineer (Worldwide Remote, Work Anywhere)

CloudLinux 51-250 IT Services

CloudLinux is hiring a Junior Infrastructure Engineer to support its remote-first infrastructure team in managing internal services, production systems, and automation across multiple environments.

Armenia Poland Portugal Bulgaria Georgia Full-time Junior Infrastructure Engineer
Ansible AWS Azure Bash CentOS CI/CD DNS GCP Git GitLab CI Jenkins JIRA Linux Python RHEL
19 hours, 13 minutes ago
Sphinx Defense

Infrastructure Deployment Manager

Sphinx Defense 1-10 Construction & Engineering

Sphinx Defense is hiring an Infrastructure Deployment Manager to build and maintain production-ready infrastructure for software development environments supporting national security programs in Space.

United States Full-time Lead Infrastructure Engineer Project Manager
$123k-$165k
Ansible Kubernetes SaltStack Terraform
1 day, 18 hours 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