Ai2

An open simulated framework for controlled experiments in robot learning

47 environments rendered with different lighting, backgrounds, table materials and camera poses. Every one of them was generated, planned and executed automatically.

Why blocks?

Kids learn a lot about space and planning from a pile of toy blocks: rebuilding a tower after it falls, turning a block over to find the letter they want. We wanted a robot learning setup with the same flavor. The domain is small enough that we know everything about every scene (each block, its letters, its pose, the goal), but the robot still has to push, flip, rotate and stack precisely, and some tasks take a dozen skills in a row.

Because nothing is hidden from us, a failed rollout is something we can actually debug. Was it the training data, the model, or the way we evaluated it? ToyBlocks is built so you can change one of those and hold the other two fixed.

85,336long-horizon demonstrations in the first release, 6–12 skills each
~1,600 hof robot execution, about 144M frames
~30Kdistinct training problems, plus 600 for validation
0teleoperated demonstrations

What's in ToyBlocks

ToyBlocks has four main parts. Each has its own page here, and every page links to the matching section of the documentation.

ToyBlocks by the numbers

Data generation

24×faster than real time on a single GPU
256environments per GPU in one Isaac Sim process in our runs, each planned separately
~250path-traced frames per second when rendering on one RTX 5090, about 10× real time
3 + 2retries and re-plans: every skill is checked against the live simulation, retried up to 3 times, then re-planned up to twice

Evaluation

50environments per eval GPU by default, with one batched model call per step
~60 minfor a 1,000-episode validation set on 7 GPUs
~144environment steps per second on a single RTX 5090 desktop GPU (28 environments)
5evaluation modes: full system, asynchronous high level, high-level oracle, low-level oracle and replay

Dataset V1

481,207skill steps in the first release, with a median plan of 5 skills and up to 18
4–10blocks per scene, 3.1–5.0 cm on a side, across 8 scene profiles
144Mframes of recorded robot data
~1,600 hof robot execution

Results from our first experiments are on the Experiments page.