From 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a Mon Sep 17 00:00:00 2001
From: YurenHao0426 <blackhao0426@gmail.com>
Date: Sat, 13 Jun 2026 12:35:36 -0500
Subject: rrm workspace: TRM/HRM/SRM code, Maze dataset, dynamical-analysis
 pipeline

Curated export for clone-and-run Maze training (2x A6000) + diagnostics.
trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible).
Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
---
 trm/requirements.txt | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 trm/requirements.txt

(limited to 'trm/requirements.txt')

diff --git a/trm/requirements.txt b/trm/requirements.txt
new file mode 100644
index 0000000..cea0f55
--- /dev/null
+++ b/trm/requirements.txt
@@ -0,0 +1,20 @@
+torch
+adam-atan2
+einops
+tqdm
+coolname
+pydantic
+argdantic
+wandb
+omegaconf
+hydra-core
+huggingface_hub
+packaging 
+ninja 
+wheel 
+setuptools 
+setuptools-scm
+pydantic-core
+huggingface_hub 
+numba
+triton
\ No newline at end of file
-- 
cgit v1.2.3