set default value to True

jax-ml · Sep 20, 2024 · 8770fb2 · 8770fb2
1 parent caf5749
commit 8770fb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/_src/config.py b/jax/_src/config.py
@@ -1530,7 +1530,7 @@ def _update_disable_jit_thread_local(val):
 
 enable_remat_opt_pass = bool_state(
     name='jax_compiler_enable_remat_pass', 
-    default=False, 
+    default=True, 
     help=('Config to enable / disable the rematerialization HLO pass. '
           'Useful to allow XLA to automatically trade off memory and '
           'compute when encountering OOM errors. However, you are '