koushikz1/arc-agi-2-jepa-episodes

ARC-AGI-2 masked-demonstration episodes (JEPA-ready) A task-level reasoning dataset built from the official ARC-AGI-2 tasks. Every row is an episode: a set of demonstration pairs (context), one held-out test_input, and the target_output the model must produce by inferring the rule shared by the demonstrations. It is not an input grid -> output grid dataset; the unit of learning is the task rule. Built 2026-09-25 with scripts/build_arc2_episodes.py (seed 20260925). Everything… See the full description on the dataset page: https://huggingface.co/datasets/koushikz1/arc-agi-2-jepa-episodes.