From 66e0d8b9fd4d0f7a2231d689c055e26fdf1cf04a Mon Sep 17 00:00:00 2001 From: YurenHao0426 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 --- env/requirements.txt | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 env/requirements.txt (limited to 'env/requirements.txt') diff --git a/env/requirements.txt b/env/requirements.txt new file mode 100644 index 0000000..d03a446 --- /dev/null +++ b/env/requirements.txt @@ -0,0 +1,13 @@ +# Merged from HRM + TRM official; versions pinned per TRM specific_requirements.txt +# torch handled separately (torch==2.7.0+cu126 via cu126 index) +einops==0.8.1 +tqdm==4.67.1 +coolname==2.2.0 +pydantic==2.11.7 +pydantic-core==2.33.2 +argdantic==1.3.3 +wandb==0.21.3 +omegaconf==2.3.0 +hydra-core==1.3.2 +huggingface_hub==0.34.4 +numba==0.61.2 -- cgit v1.2.3