env/requirements.txt


1
2
3
4
5
6
7
8
9
10
11
12
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