Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,9 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
datasets:
|
| 4 |
+
- emiliocantuc/sudoku-extreme-1k-aug-1000
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
A simple JAX reproduction of the Tiny Recursive Model (TRM) trained on the sudoku task. Code [here](https://github.com/ecntu/trm-jax).
|
| 8 |
+
|
| 9 |
+
Training metrics included (see the "Training Metrics" tab).
|