---
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).