diff --git a/env/hpc/modules.txt b/env/hpc/modules.txt
index 0617883..1db82fb 100644
--- a/env/hpc/modules.txt
+++ b/env/hpc/modules.txt
@@ -1,3 +1,4 @@
+PyTorch/2.1.2-foss-2023a-CUDA-12.1.1
 jax/0.4.25-gfbf-2023a-CUDA-12.1.1
 Flax/0.8.4-gfbf-2023a-CUDA-12.1.1
 Optax/0.2.2-gfbf-2023a-CUDA-12.1.1