RemoteLeaf Logo
Log in Sign up

Senior Machine Learning Engineer, ML Infrastructure - Offline

1 month, 2 weeks ago
China
Full-time
Senior
Machine Learning Engineer
Software Development
Apache Airflow Apache Spark Microservices 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 125

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 technologies such as PyTorch, Ray Data, and Ray Train.
  • Integrate ML pipelines with workflow orchestration systems such as Flyte, Airflow, or similar tools to enable reliable multi-stage training workflows.
  • Improve reproducibility and observability of ML pipelines 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 enable efficient large-scale experimentation and model iteration.
  • Lead architectural improvements to ensure offline ML pipelines remain scalable, reliable, and cost-efficient.
  • Shape how model datasets are prepared, validated, and delivered to distributed training systems.
  • Support production ML systems by maintaining reliable infrastructure for training workflows.
  • Work across ML engineering and platform teams to handle growing data volumes and increasingly complex training workloads.

Requirements

  • Experience working with distributed computing frameworks such as Ray, Spark, or Flink, with familiarity in the Ray ecosystem including Ray Data and Ray Train.
  • Experience building and optimizing large-scale distributed ML training pipelines with Torch Compilation, Quantization, CUDA, or GPU kernel optimization.
  • 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 working with large-scale distributed workloads.
  • Experience with modern data infrastructure such as 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.
  • Experience with workflow orchestration systems such as Flyte or Airflow, or similar tools.
  • Strong English communication skills for frequent written and verbal collaboration with global colleagues and partners.
  • Relocation support is not available for this position.
  • Work visa or immigration sponsorship is not available for this position.

Benefits

  • Comprehensive health, life, and disability insurance.
  • Commute subsidy.
  • 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.
  • Training and development programs.
  • Office food snacks.
  • Employee Resource Groups.
  • Global Employee Assistance Program.
  • Volunteering and donation matching program.

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
4 hours, 41 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
4 hours, 41 minutes ago
Loka

Machine Learning Engineer

Loka 51-250 Internet Software & Services

Loka is hiring a remote Machine Learning Engineer to build, deploy, and improve ML solutions for clients across startups and life sciences.

Serbia North Macedonia Portugal Latin America Full-time Junior Machine Learning Engineer
AWS Deep Learning Generative AI Hugging Face LLM Machine Learning MLflow MLOps NLP Python PyTorch TensorFlow
1 day, 3 hours ago
Databricks

Sr. Specialist Solutions Architect

Databricks 1K-5K IT Services

Databricks is hiring a Senior Specialist Solutions Architect (ML & AI) in London to help enterprise customers design and deploy production-grade machine learning and AI solutions on its Data Intelligence Platform.

United Kingdom Full-time Lead Machine Learning Engineer Solutions Engineer
Apache Spark AWS Azure GCP Generative AI LLM Machine Learning MLOps
1 day, 3 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