Blog Your GPUs Are Everywhere. Your Robot-Learning Loop Shouldn't Be.
MLOps 8 mins

Your GPUs Are Everywhere. Your Robot-Learning Loop Shouldn't Be.

Robotics compute is spreading across clouds and clusters. Learn how one portable pipeline layer can keep the robot-learning loop reproducible.

ZenML blog cover with the title 'Your GPUs Are Everywhere. Your Robot-Learning Loop Shouldn't Be.' next to a geometric isometric illustration in blue, red, and cream
On this page

Robotics teams do not have a training job. They have a loop.

A robot records camera frames, joint states, actions, and failures. That data is cleaned and turned into episodes. A policy is trained, evaluated in simulation, tested on hardware, and eventually deployed. The next day, the fleet produces new evidence and the loop begins again.

Each part wants different infrastructure. Data processing belongs near the data. Simulation may need a large CPU and GPU fleet. Training wants the fastest accelerators the team can obtain. Hardware evaluation happens in the lab. Deployment ends at the edge.

That is why the defining infrastructure problem in robotics is not simply getting a large training run to finish. It is keeping the whole learning loop coherent while the compute underneath it keeps changing.

Robot learning has outgrown a single cluster

The upper end of the market makes the scale visible. NVIDIA reports that GR00T N1 used up to 1,024 H100 GPUs for a single model and roughly 50,000 H100 GPU-hours for pretraining. The cluster was managed by OSMO, NVIDIA’s orchestration platform for robotics workloads (GR00T N1 technical report).

Most robotics companies will not train at NVIDIA’s scale. But smaller teams face a messier version of the same systems problem. Capacity may come from a hyperscaler agreement, an AI cloud, an on-premises Kubernetes cluster, or a workstation in the lab. The mix changes as credits expire, new hardware arrives, and model requirements grow.

The strongest evidence comes from the teams themselves. Dyna Robotics is hiring an ML infrastructure engineer to turn its “multi-cloud GPU fleet into a training engine” where every run is reproducible and a researcher’s next experiment is one command away (Dyna Robotics job posting). That is the problem in one sentence: the GPUs exist, but the team still needs a reliable way to use them as one system.

Cloud vendors see the same demand. CoreWeave now has a physical AI offering for simulation, policy training, VLA fine-tuning, and sim-to-real validation. Nebius and NVIDIA have launched a managed physical AI stack built around OSMO, Cosmos, and Nebius infrastructure. This is more than a new cloud category. It is a signal that robotics infrastructure has become a market of its own.

The expensive failure is a broken handoff

GPU utilization matters, but utilization is downstream of the workflow.

If data preparation, simulation, training, evaluation, and deployment are separate scripts owned by separate people, every transition becomes a place to lose context. Which episode set produced this checkpoint? Which preprocessing code ran? Was the failed evaluation caused by a model change, a simulator change, or a different dataset? Can the team reproduce the result on another backend next week?

RoboForce described the operational cost plainly: “Manual handoffs between data generation, simulation, and training means our GPUs can sit idle.” After moving the workflow behind a single configuration, the company says its iteration cycle fell from weeks to days (Nebius and NVIDIA customer announcement). It is a vendor case study, so the numbers deserve the usual caution. The workflow problem it describes is nevertheless familiar: idle compute is often a symptom of disconnected stages, not a lack of schedulers.

The data side is just as demanding. Wayve describes a fleet-learning loop that continuously collects, curates, trains, re-simulates, and validates models. Its public architecture evolved from an office server to custom ingest stations, Kubernetes-based simulation, Apache Beam preprocessing, and Azure storage capable of feeding production training from petabyte-scale datasets (Wayve’s engineering account). The hard part is not any one tool. It is maintaining the contract between all of them.

Three reasonable strategies — and their trade-offs

There is no universal answer to this problem. Robotics teams generally choose some combination of three strategies.

Standardize on one infrastructure vendor. This reduces integration work and can be the right choice when capacity, pricing, and geography line up. Managed offerings from NVIDIA, Nebius, CoreWeave, and the hyperscalers are increasingly capable. The trade-off is that workflow decisions can become coupled to one provider’s execution model.

Build a platform in-house. At sufficient scale, this is rational. Wayve built a sophisticated fleet-learning platform because its data volume and autonomy workflow justified the investment. The cost is not only the initial build: a platform team must keep adapting it as researchers, models, clusters, and deployment targets change.

Add a portable pipeline layer above the infrastructure. This does not replace Kubernetes, SageMaker, Vertex AI, or a cluster scheduler. It defines the workflow independently and delegates execution to those systems. The team gets one representation of the learning loop while retaining the ability to change where it runs.

The third strategy is most useful for the awkward middle: teams with more than one useful compute environment, but without the appetite to build and maintain an internal orchestration product.

What a control plane must preserve

A credible control plane for robot learning needs to preserve five things as infrastructure changes:

  1. The pipeline. Episode ingestion, preprocessing, training, evaluation, and release should have one explicit dependency graph instead of a chain of scripts and messages.
  2. Execution choice. The pipeline definition should stay stable while configured infrastructure determines where it runs. Individual high-compute steps should be able to use a specialized environment when necessary.
  3. Artifacts and lineage. A checkpoint should point back to its producing run, parameters, code, and upstream artifacts. External datasets still need deliberate versioning, but the workflow should not discard their identity at each handoff.
  4. Repeatability without waste. When code, parameters, and inputs are unchanged, safe steps should reuse previous outputs. Steps that read mutable external systems must opt out or use an explicit cache policy.
  5. Automation with gates. Schedules and external events should be able to start the loop, while evaluation thresholds and human review control what reaches a robot.

These requirements are deliberately above the scheduler. A scheduler decides which node runs a job. A pipeline control plane records why that job exists, what it consumes, what it produces, and what should happen next.

Where ZenML fits

ZenML is that pipeline layer. It does not sell GPU capacity or replace the systems that schedule it. It lets a team express the learning loop as Python steps and map those steps onto a configured stack.

ZenML supports orchestrators including Kubernetes, Kubeflow, SageMaker, Vertex AI, AzureML, and SkyPilot. Changing the active stack changes the execution backend without requiring a second copy of the pipeline. Backend-specific credentials, images, storage, and settings still have to be configured; portability is an architectural boundary, not magic.

Step operators provide a second level of control. A pipeline can run under its normal orchestrator while a training step is submitted to a specialized environment such as SageMaker, Vertex AI, AzureML, Kubernetes, Modal, or Run:AI. A shared artifact store carries the inputs and outputs across that boundary.

Around execution, ZenML records pipeline runs and artifacts, supports configurable step caching, and can be invoked by schedules or external systems. For a robotics workflow, that means the episode batch, processed dataset, policy checkpoint, evaluation report, and release decision can remain parts of one traceable run rather than five disconnected jobs.

The result is not “one giant cluster.” It is one learning loop that can survive several clusters.

When this approach is not the right one

If all of your workloads fit comfortably in one managed platform, adding another abstraction may not pay for itself. If your company has the scale and platform-engineering depth to build a robotics-specific system like OSMO, owning every layer may be a strategic advantage. And no workflow tool will make a tightly coupled distributed training job span arbitrary clouds with poor network links; that job still needs an appropriate cluster.

But if your data, simulation, training, and evaluation workloads already cross infrastructure boundaries, those boundaries should not leak into every pipeline. Researchers should be able to improve the policy without relearning the deployment topology. Platform engineers should be able to change the topology without rewriting the science.

That is the opportunity for a neutral control plane: not to hide the real differences between clouds and clusters, but to keep those differences from breaking the robot-learning loop.

If that describes your team, book a demo or try the open-source framework.

Start deploying AI workflows in production today

Enterprise-grade AI platform trusted by thousands of companies in production

Continue Reading

Why Retail MLOps Is Harder Than You Think

Why Retail MLOps Is Harder Than You Think

An in-depth analysis of retail MLOps challenges, covering data complexity, edge computing, seasonality, and multi-cloud deployment, with real-world examples from major retailers like Wayfair and Starbucks, and practical solutions including ZenML's impact in reducing deployment time from 8.5 to 2 weeks at Adeo Leroy Merlin.