diff --git a/conf/jax.config b/conf/jax.config
index 7cf790e..56c0f12 100644
--- a/conf/jax.config
+++ b/conf/jax.config
@@ -19,7 +19,7 @@ singularity{
     cacheDir = params.singularity_cache_dir
 }
 params {
-      max_memory = 768.GB
-      max_cpus = 70
+      max_memory = 320.GB
+      max_cpus = 32
       max_time = 336.h
     }