diff --git a/env/hpc/modules.txt b/env/hpc/modules.txt
index c780513..4c4a1d7 100644
--- a/env/hpc/modules.txt
+++ b/env/hpc/modules.txt
@@ -1,6 +1,3 @@
-gfbf/2024a
 GCCcore/13.3.0
 Python/3.12.3-GCCcore-13.3.0
-PyTorch/2.7.1-foss-2024a-CUDA-12.6.0
 FFmpeg/7.0.2-GCCcore-13.3.0
-PyYAML/6.0.2-GCCcore-13.3.0
diff --git a/env/hpc/requirements.txt b/env/hpc/requirements.txt
index eeca95a..5314edb 100644
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@@ -4,8 +4,8 @@ evosax==0.2.0
 flax>=0.12.2
 gymnasium>=1.2.3
 ipykernel==7.2.0
-jax[cuda12_local]==0.9.0.1
-scipy>=1.14.0
+jax[cuda13]==0.9.0.1
+torch
 numpy>=2.0.0
 protobuf>=5.0.0
 warp-lang
diff --git a/scripts/hpc/install.sh b/scripts/hpc/install.sh
index 541af16..2c4d12c 100644
--- a/scripts/hpc/install.sh
+++ b/scripts/hpc/install.sh
@@ -44,11 +44,6 @@ source vsc-venv --activate \
 set -euo pipefail
 cd "$PBS_O_WORKDIR"
 
-# Prepend venv to PYTHONPATH to shadow system packages (e.g. SciPy-bundle compiled
-# against NumPy 1.x) that are injected by the loaded modules.
-VENV_LIB_DIR="$VIRTUAL_ENV/lib/python$(python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')/site-packages"
-export PYTHONPATH="$VENV_LIB_DIR:${PYTHONPATH:-}"
-
 echo '>>> Installing ipykernel...'
 CLUSTER_ID="${VSC_INSTITUTE_CLUSTER:-generic}"
 python -m ipykernel install --user --name="sel3_${CLUSTER_ID}" \
diff --git a/scripts/hpc/train.pbs b/scripts/hpc/train.pbs
index 1610920..b4ae77c 100644
--- a/scripts/hpc/train.pbs
+++ b/scripts/hpc/train.pbs
@@ -50,11 +50,6 @@ source vsc-venv --activate \
 set -euo pipefail
 cd "$PBS_O_WORKDIR"
 
-# Prepend venv to PYTHONPATH to shadow system packages (e.g. SciPy-bundle compiled
-# against NumPy 1.x) that are injected by the loaded modules.
-VENV_LIB_DIR="$VIRTUAL_ENV/lib/python$(python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')/site-packages"
-export PYTHONPATH="$VENV_LIB_DIR:${PYTHONPATH:-}"
-
 
 echo ">>> Starting BrittleStar training..."
 export MUJOCO_GL=egl