diff --git a/eb_hooks.py b/eb_hooks.py
index e7d31398..00b4ea2c 100644
--- a/eb_hooks.py
+++ b/eb_hooks.py
@@ -2195,4 +2195,7 @@ def set_maximum(parallel, max_value):
     'ROCm-LLVM': {
         '*': (set_maximum, 12),
     },
+    'jax': {
+        '*': (divide_by_factor, 2),
+    },
 }