Models
Datasets
Spaces
Buckets new
Docs
Enterprise
Pricing
- Website
- Community
- Solutions
Log In
Sign Up

rogermt
/

ARC-AGI

Model card Files Files and versions

ARC-AGI / trm_solver /trm_nano

43.1 kB

Ctrl+K

Ctrl+K

2 contributors

History: 13 commits

rogermt's picture

Fix: total_mem -> total_memory"

e9d6928 verified 2 months ago

README.md

3.19 kB
Add TRM-Nano README with architecture details and usage" 2 months ago
model.py

7.66 kB
Add TRM-Nano model implementation (h=96, 4 heads, 2 layers, unrolled recursion)" 2 months ago
onnx_builder.py

8.8 kB
Fix: add build_trm_nano_onnx function (alias used by train_task.py) 2 months ago
quick_test.py

4.85 kB
Fix: total_mem -> total_memory" 2 months ago
solved_tasks.json

244 Bytes
Add solved_tasks.json - 51 tasks solved by analytical solvers 2 months ago
train_task.py

18.4 kB
Fix: force unbuffered stdout for Kaggle notebook output + use print() with flush" 2 months ago