From 3e586da95387417c95a1d81e582e0bd0878fab1b Mon Sep 17 00:00:00 2001
From: Tibo De Peuter <tibo.depeuter@ugent.be>
Date: Sun, 5 Apr 2026 00:22:04 +0200
Subject: [PATCH] fixup! fix(hpc): rework modules again

---
 env/hpc/requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/env/hpc/requirements.txt b/env/hpc/requirements.txt
index 20a1fe1..eeca95a 100644
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@@ -4,7 +4,7 @@ evosax==0.2.0
 flax>=0.12.2
 gymnasium>=1.2.3
 ipykernel==7.2.0
-jax[cuda13]==0.9.0.1
+jax[cuda12_local]==0.9.0.1
 scipy>=1.14.0
 numpy>=2.0.0
 protobuf>=5.0.0