ARC Diffusion

A diffusion approach to solving ARC.

Quickstart: To get started on Runpod, you can avail of this one-click template (affiliate link).

Running scripts: Install with

uv sync

and see how to run scripts by running:

uv run script.py -h

e.g.

uv run pipeline.py -h

You can run on mac or cpu or gpu (cuda).

Some sample scripts: To run training and evaluation (and pushing to hf):

uv run pipeline.py --config outputs/test_config.json

To run evaluation, with majority voting:

uv run evaluate.py --config outputs/test/config.json --maj --stats --prefer-best