1 files changed, 20 insertions, 0 deletions
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