File size: 270 Bytes
fdb9bd0 | 1 2 3 4 5 6 7 8 9 | ---
license: mit
datasets:
- emiliocantuc/sudoku-extreme-1k-aug-1000
---
A simple JAX reproduction of the Tiny Recursive Model (TRM) trained on the sudoku task. Code [here](https://github.com/ecntu/trm-jax).
Training metrics included (see the "Training Metrics" tab). |