From 745c09e5245dbc498bc796aeec19682011733c82 Mon Sep 17 00:00:00 2001 From: Tibo De Peuter Date: Sat, 4 Apr 2026 20:50:07 +0200 Subject: [PATCH] fix(hpc): add PyTorch module to minimize uv --- env/hpc/modules.txt | 1 + 1 file changed, 1 insertion(+) 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