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