ARC-AGI / trm_solver

Commit History

  1. Fix: total_mem -> total_memory"

    Verified signature
    e9d6928
  2. Add quick_test.py - standalone timing test, prints immediately, no imports from other files"

    Verified signature
    27d7101
  3. Fix: force unbuffered stdout for Kaggle notebook output + use print() with flush"

    Verified signature
    0e4cf22
  4. Add --log_batch_every: prints loss every N batches within an epoch"

    Verified signature
    e997a38
  5. Add --num_gpus: splits tasks across GPUs in parallel processes"

    Verified signature
    626c4f7
  6. Add --log_every arg to control epoch print frequency (default 100)

    Verified signature
    e0b6ea4
  7. Fix: add build_trm_nano_onnx function (alias used by train_task.py)

    Verified signature
    492f908
  8. Add solved_tasks.json - 51 tasks solved by analytical solvers

    Verified signature
    a917394
  9. Replace train_task.py with full batch runner: loops all 400 tasks, skips solved, logs everything, uses manual ONNX builder

    Verified signature
    28954c6
  10. Add manual ONNX builder with weight reuse - fits 1.44MB with full h=96 T=3 n=4"

    Verified signature
    ab04ef8
  11. Add TRM-Nano README with architecture details and usage"

    Verified signature
    70b26ba
  12. Add TRM-Nano per-task training script with augmentation and ONNX export

    Verified signature
    7dd4c24
  13. Add TRM-Nano model implementation (h=96, 4 heads, 2 layers, unrolled recursion)"

    Verified signature
    441b2d5

  1. Update classifier to support Kilo local server

    Verified signature
    ccffc8f
  2. Update TODO with TRM findings, fix outdated info

    Verified signature
    826a998
  3. Add TRM research findings and architecture notes

    Verified signature
    0e63a4d
  4. Add composition solvers for chained transforms

    Verified signature
    981ef11
  5. Add DeepSeek task classifier for LLM-guided solver routing

    Verified signature
    9caa30d
  6. reorg kilo skills folder

    Roger MT
    9ae02a4
  7. Add TRM solver skill for agent guidance

    Verified signature
    cd8c1ce
  8. Upload trm_solver/__init__.py

    Verified signature
    e4e4da5
  9. Add TRM solver README

    Verified signature
    51d8a6c
  10. Add Kilo Bridge for LLM-driven ARC task analysis and ONNX export

    Verified signature
    d565d28
  11. Add TRM solver NN executor with 14 transform types and ONNX export

    Verified signature
    eea0011