TL;DR
- NVIDIA published technical guidance on post-training Cosmos 3 Edge, a 4B-parameter world model with a 2B Nemotron-based reasoner, built to run robot control directly on Jetson Thor hardware instead of a data center.
- The model was fine-tuned on Cosmos3-DROID, a dataset of 76,000 teleoperated robot trajectories spanning about 350 hours across 86 tasks and 564 scenes.
- On a Jetson AGX Thor T5000, the trained policy plans roughly 1.53 seconds of action per compute chunk to cover about 2.13 seconds of real robot motion.
- In closed-loop RoboLab testing, the resulting policy hit a 22.9% success rate, a number that says this is early-stage engineering, not a finished product.
A Robot Brain Small Enough to Fit in the Robot
NVIDIA published a technical writeup on post-training Cosmos 3 Edge, a compact world model built for robots that can’t phone home to a data center every time they need to plan a move. The company describes it plainly: Cosmos 3 Edge is a 4B omni-model with a 2B NVIDIA Nemotron-based reasoner in the Cosmos 3 family. That reasoner does the actual thinking about physical space, and NVIDIA sized the whole package to fit inside the memory budget of a Jetson Thor chip.
To teach it robot manipulation, NVIDIA fine-tuned the model on Cosmos3-DROID, built from 76,000 successful teleoperated trajectories. That’s roughly 350 hours of a human operator guiding a robot arm through 86 distinct tasks across 564 different scenes. Getting the base model ready for that fine-tuning wasn’t a small job either: NVIDIA’s writeup logs 64 training nodes, 60,000 iterations, and about 17,400 GB200-hours of compute before the edge-ready version ever existed.
Once trained, the payoff is speed. On a Jetson AGX Thor T5000, the policy generates roughly 1.53 seconds of action planning per compute chunk, which has to cover about 2.13 seconds of actual robot motion. In closed-loop RoboLab testing, that policy reached a 22.9% success rate.
The Math Behind a 4B Robot Brain
Call this the honest number in an industry that rarely publishes one. A 22.9% success rate isn’t the kind of headline you’d expect from a company as confident as NVIDIA usually sounds, and that’s exactly why it’s worth sitting with. Every prior generation of world models has hit the same wall: the more physical knowledge you cram in, the smarter the predictions get, but the harder the model becomes to run anywhere except a rack of data-center GPUs. Cosmos 3 Edge tries to punch through that wall by shrinking the model instead of shrinking the ambition.
Think of it like fitting a full orchestra’s sheet music onto a single busker’s folding stand. You can’t bring the whole ensemble, so you rewrite the score down to what one musician can actually play, and you accept some of the texture is gone. NVIDIA did something similar here. They took reasoning capacity that normally lives in a much larger world model and repackaged it into a 4B-parameter model with a 2B reasoner, small enough to sit inside a Jetson Thor’s memory without a cloud round-trip for every decision. A bet, not a certainty.
The 1.53-second planning window covering 2.13 seconds of motion is the part I keep coming back to. I’ll admit it surprised me more than the parameter count did. That’s a robot’s control system staying just barely ahead of its own body, the way a simultaneous interpreter translates a sentence before the speaker has finished it. Miss that window and the robot either freezes or acts on stale information. Would you trust a warehouse robot moving near you if it succeeded barely one time in five during lab testing?
NVIDIA frames this against data-center GPU offloading approaches, the more common route where a robot sends sensor data out and waits for a cloud model to send commands back. That round-trip adds latency no factory floor or warehouse aisle can tolerate. Cosmos 3 Edge is a bet that on-device is worth the accuracy tradeoff, at least for now. Is 22.9% a floor NVIDIA climbs from quickly, or a ceiling that takes years to move? NVIDIA doesn’t answer that in this post, and I don’t think it can yet.
Why World Models Usually Skip the Edge
World foundation models exist because teaching a robot physics from scratch, every single time, for every single task, is a losing game. Give a model broad physical knowledge upfront, the theory goes, and it needs far less task-specific training to become useful. The catch NVIDIA’s own background material points to is plain: that knowledge usually comes packaged at a size with nowhere to run except a server rack.
Jetson Thor is NVIDIA’s answer to the hardware half of that problem, an edge compute platform built for exactly this kind of onboard robotics workload. Pairing it with a model purpose-built to fit inside its memory, rather than a shrunk-down version of something bigger, is the more interesting choice here. NVIDIA didn’t just compress an existing model. It built the Cosmos 3 Edge tier as part of the family from the start, with the 2B reasoner sized specifically for that constraint.
None of this happens in a vacuum. Robotics teams building on Jetson hardware have been asking for on-device intelligence for a while, mostly because latency from cloud offloading kills the use case for anything that needs to react in real time, a warehouse robot avoiding a person who just stepped into its path, for instance. Does a robot working near people really need cloud-level model accuracy, or just enough certainty not to hurt someone? Whether Cosmos 3 Edge’s current accuracy clears that bar is a separate question from whether the architecture itself is right, and NVIDIA seems to know it, given how upfront the 22.9% figure is.
What Comes Next for Cosmos 3 Edge
A few things worth watching from here. First, whether that 22.9% RoboLab success rate moves as NVIDIA expands Cosmos3-DROID past its current 86 tasks and 564 scenes, since more varied training data is the obvious next lever to pull. Second, whether independent robotics teams running Cosmos 3 Edge on their own Jetson Thor hardware report numbers anywhere close to NVIDIA’s internal benchmark, because self-reported closed-loop results from a model’s own creator always deserve a second look.
And third, whether NVIDIA extends this edge-sized treatment to other members of the Cosmos 3 family, or whether Cosmos 3 Edge stays a one-off experiment aimed narrowly at Jetson Thor. If the compute-versus-capability tradeoff keeps improving even modestly, expect this to become the template for how NVIDIA talks about robot AI going forward, not the exception.
Editor's Note
What stands out to me is NVIDIA publishing a 22.9% success rate without dressing it up. I've seen plenty of robotics announcements bury the hard number three paragraphs deep, if at all. This one leads with it. I'm less interested in the parameter count than in that 1.53 second planning window, because that's the actual constraint robots live under. My guess: this number climbs fast once more teams start testing outside NVIDIA's own lab.
– Sanket Chaukiyal, founder, SmartChunks
FAQ
What is Cosmos 3 Edge?
It is a 4 billion parameter omni-model from NVIDIA's Cosmos 3 family, built with a 2 billion parameter Nemotron-based reasoner and sized specifically to run on-device for robot control instead of in a data center.
How accurate is the post-trained Cosmos 3 Edge model?
In closed-loop RoboLab testing, NVIDIA's post-trained policy reached a 22.9% success rate, a number NVIDIA published directly rather than rounding away.
What hardware does Cosmos 3 Edge run on?
NVIDIA tested it on a Jetson AGX Thor T5000, where the policy generates about 1.53 seconds of action planning per compute chunk to cover roughly 2.13 seconds of real robot motion.
What data was used to train it?
NVIDIA post-trained the model on Cosmos3-DROID, a dataset of 76,000 successful teleoperated robot trajectories spanning about 350 hours, 86 tasks, and 564 scenes.
