├── LICENSE ├── README.md ├── assets ├── hrm.png └── npyjs.js ├── config ├── arch │ ├── hrm_v1.yaml │ └── hrm_v2_params_matched.yaml └── cfg_pretrain.yaml ├── dataset ├── build_arc_dataset.py ├── build_maze_dataset.py ├── build_sudoku_dataset.py └── common.py ├── evaluate_trained_model.py ├── evaluators ├── arc.py └── arc_augmentation_ablation.py ├── kaggle └── input │ ├── arc-agi_concept-challenges.json │ ├── arc-agi_concept-solutions.json │ ├── arc-agi_evaluation-challenges.json │ ├── arc-agi_evaluation-solutions.json │ ├── arc-agi_training-challenges.json │ ├── arc-agi_training-solutions.json │ ├── checksum.md │ └── sample_submission.json ├── models ├── common.py ├── hrm │ ├── hrm_act_v1.py │ └── hrm_act_v2.py ├── layers.py ├── losses.py └── sparse_embedding.py ├── pretrain.py ├── puzzle_dataset.py ├── requirements.txt ├── run_augmentation_ablation_eval.py └── utils └── functions.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/README.md -------------------------------------------------------------------------------- /assets/hrm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/assets/hrm.png -------------------------------------------------------------------------------- /assets/npyjs.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/assets/npyjs.js -------------------------------------------------------------------------------- /config/arch/hrm_v1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/config/arch/hrm_v1.yaml -------------------------------------------------------------------------------- /config/arch/hrm_v2_params_matched.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/config/arch/hrm_v2_params_matched.yaml -------------------------------------------------------------------------------- /config/cfg_pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/config/cfg_pretrain.yaml -------------------------------------------------------------------------------- /dataset/build_arc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/dataset/build_arc_dataset.py -------------------------------------------------------------------------------- /dataset/build_maze_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/dataset/build_maze_dataset.py -------------------------------------------------------------------------------- /dataset/build_sudoku_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/dataset/build_sudoku_dataset.py -------------------------------------------------------------------------------- /dataset/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/dataset/common.py -------------------------------------------------------------------------------- /evaluate_trained_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/evaluate_trained_model.py -------------------------------------------------------------------------------- /evaluators/arc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/evaluators/arc.py -------------------------------------------------------------------------------- /evaluators/arc_augmentation_ablation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/evaluators/arc_augmentation_ablation.py -------------------------------------------------------------------------------- /kaggle/input/arc-agi_concept-challenges.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/arc-agi_concept-challenges.json -------------------------------------------------------------------------------- /kaggle/input/arc-agi_concept-solutions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/arc-agi_concept-solutions.json -------------------------------------------------------------------------------- /kaggle/input/arc-agi_evaluation-challenges.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/arc-agi_evaluation-challenges.json -------------------------------------------------------------------------------- /kaggle/input/arc-agi_evaluation-solutions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/arc-agi_evaluation-solutions.json -------------------------------------------------------------------------------- /kaggle/input/arc-agi_training-challenges.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/arc-agi_training-challenges.json -------------------------------------------------------------------------------- /kaggle/input/arc-agi_training-solutions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/arc-agi_training-solutions.json -------------------------------------------------------------------------------- /kaggle/input/checksum.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/checksum.md -------------------------------------------------------------------------------- /kaggle/input/sample_submission.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/kaggle/input/sample_submission.json -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/models/common.py -------------------------------------------------------------------------------- /models/hrm/hrm_act_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/models/hrm/hrm_act_v1.py -------------------------------------------------------------------------------- /models/hrm/hrm_act_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/models/hrm/hrm_act_v2.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/models/layers.py -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/sparse_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/models/sparse_embedding.py -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/pretrain.py -------------------------------------------------------------------------------- /puzzle_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/puzzle_dataset.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_augmentation_ablation_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/run_augmentation_ablation_eval.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arcprize/hierarchical-reasoning-model-analysis/HEAD/utils/functions.py --------------------------------------------------------------------------------