Senior Machine Learning Engineer, ML Infrastructure - Offline

1 month ago
Full-time
Senior
Software Development
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

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

AI/Machine Learning Engineer

Moonshot 51-250 Diversified Consumer Services

Moonshot is hiring a remote AI/Machine Learning Engineer in Ireland to help build its development team and advance software that counters online harms through model development, integration, and analysis.

Agile AWS CI/CD GCP Kanban Machine Learning MySQL NLP Python PyTorch Scikit-learn Scrum
22 hours, 1 minute ago

GTM Operations Engineer

K1x 51-250 Internet Software & Services

K1x is hiring a GTM Operations Engineer to build and scale AI-driven automation across the go-to-market and post-sale customer lifecycle for its tax software business.

Databricks HubSpot Salesforce Snowflake
22 hours, 31 minutes ago

AI/ML Engineer

GlobalDev Tech 51-250 Internet Software & Services

Axonlab is hiring an AI/Data Science professional to design and deliver end-to-end machine learning and computer vision solutions for its pan-European medtech eCommerce platform.

Azure Computer Vision Feature Engineering Machine Learning MLOps NLP NumPy Pandas Python PyTorch Scikit-learn TensorFlow
22 hours, 46 minutes ago

Engineering Tech Lead

TTEC Digital 1K-5K Professional Services

TTEC Digital is hiring a fully remote Engineering Technical Lead to help build and lead a real-time AI customer experience platform for enterprise clients.

Agile CRM GCP Generative AI Go Kafka LLM Machine Learning NATS Prototyping Python WebSockets
22 hours, 46 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