comment clear_jax memory function (#83)

Author: rahul-anand

Ironwood/src/benchmark_utils.py

@@ -169,7 +169,9 @@ def multiple_iteration_timeit_from_trace(
 
                     result = compute_func(*data_args)
                     jax.block_until_ready(result)
-            clear_jax_memory()
+
+            # Commenting it out as it's causing issues with GEMM
+            # clear_jax_memory()
     trace = get_trace(tmp_trace_dir)
 
     if trace_full_dir != tmp_trace_dir: