diff --git a/kmem.sh b/kmem.sh
index f8609a1..24dd2b7 100755
--- a/kmem.sh
+++ b/kmem.sh
@@ -46,7 +46,7 @@
 TARGET=${TARGET:-riscv32-unknown-elf}
 
 # Default is a summary of release build.
-DETAILS="0"
+DETAILS=""
 BUILD="release"
 
 function parseargv {
@@ -58,12 +58,12 @@
     for i; do
         case "$1" in
             -d|--details)
-                DETAILS="1"
+                DETAILS="--details"
                 shift
                 ;;
 
             -s|--summary)
-                DETAILS="0"
+                DETAILS=""
                 shift
                 ;;
 
@@ -93,4 +93,5 @@
 parseargv "$@"
 
 CANTRIP_OUT="${ROOTDIR}/out/cantrip/${TARGET}/${BUILD}"
-exec awk -f "${ROOTDIR}/scripts/mem.awk" "${CANTRIP_OUT}/system.cdl" DETAILS="${DETAILS}"
+PYTHONPATH="${PYTHONPATH}:${ROOTDIR}/cantrip/projects/capdl/python-capdl-tool"
+exec python3 "${ROOTDIR}/cantrip/tools/seL4/kmem-tool/kmem.py" --object-state "${CANTRIP_OUT}/object-final.pickle" ${DETAILS}
diff --git a/mem.awk b/mem.awk
deleted file mode 100644
index 429da6c..0000000
--- a/mem.awk
+++ /dev/null
@@ -1,127 +0,0 @@
-#
-# Copyright 2022 Google LLC
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-# seL4 capDL spec memory analyzer.
-
-BEGIN {
-  in_objects = 0;
-  frame_types[1] = "elf";
-  frame_types[2] = "bss";
-  frame_types[3] = "ipc_buffer";
-  frame_types[4] = "stack";
-  frame_types[5] = "copyregion";
-  frame_types[6] = "bootinfo";
-  frame_types[7] = "mmio";
-  DETAILS = 0;
-}
-/^objects.*{/ { in_objects = 1; }
-in_objects && $3 == "frame" {
-  component = $1;
-  size = 0;
-  switch (substr($4, 2)) {
-    case /4k/: size = 4096; break;
-    case /4M/: size = 4*1024*1024; break;
-    default:
-      print "Unknown frame size", substr($4, 2);
-      break;
-  }
-
-  switch ($8) {
-    case "CDL_FrameFill_BootInfo":
-      split(component, a, "_");
-      component = a[3] "_" a[4];
-      frame_type = "bootinfo";
-      break;
-    case "CDL_FrameFill_FileData":
-      sub(".*frame_", "", component);
-      sub("_group_bin.*", "", component);
-      frame_type = "elf";
-      break;
-    default: {
-      switch (component) {
-        case /_copy_region_/:
-          sub("_copy_region.*", "", component);
-          frame_type = "copyregion";
-          break;
-        case /_frame__camkes_ipc_buffer_/:
-          sub("_frame__camkes_ipc_buffer.*", "", component);
-          frame_type = "ipc_buffer";
-          break;
-        case /^stack__camkes_stack_/:
-          sub("^stack__camkes_stack_", "", component);
-          split(component, a, "_");
-          component = a[1] "_" a[2];
-          frame_type = "stack";
-          break;
-        case /_data_[0-9]_obj/:
-          sub("_?[0-9]*_data_[0-9]_obj", "", component);
-          sub("_mmio.*", "", component);
-          frame_type = ($5 == "paddr:" ? "mmio" : "bss");
-          break;
-        default:
-          if ($5 == "paddr:") {
-            sub("_mmio.*", "", component);
-            sub("_csr_.*", "", component);
-            frame_type = "mmio";
-          } else {
-            sub(".*frame_", "", component);
-            sub("_group_bin.*", "", component);
-            frame_type = "bss";
-          }
-          break;
-      }
-      break;
-    }
-  }
-  cur_component = component;
-  if (frame_type != "copyregion" && frame_type != "mmio") {
-    # NB: copyregion's are holes in the VSpace
-    components[cur_component] += size;;
-  }
-  memory[cur_component, frame_type] += size;
-}
-in_objects && /^}/ { in_objects = 0; }
-
-END {
-  asorti(components, comps, "@ind_str_asc");
-  total = 0;
-  for (i in comps) {
-    c = comps[i];
-    ntypes = 0;
-    # per-component breakdown by frame type
-    for (j in frame_types) {
-      t = frame_types[j];
-      if (memory[c, t] != "") {
-        if (DETAILS) {
-          printf "%-36.36s %-14s %5.0f KiB\n", (ntypes > 0 ? "" : c), t, memory[c, t] / 1024.;
-        }
-        ntypes++;
-      }
-    }
-    total += components[c];
-    # per-component totals
-    if (!DETAILS) {
-        printf "%-36.36s %5.0f KiB (%s)\n", c, components[c] / 1024., components[c];
-    } else if (ntypes > 1) {
-        printf "%-36.36s %-14s %5.0f KiB (%s)\n", c, "total", components[c] / 1024., components[c];
-    }
-  }
-  # overall total
-  if (DETAILS) {
-      printf "%-36.36s %-14s %5.0f KiB (%s)\n", "Total", "", total / 1024., total;
-  } else {
-      printf "%-36.36s %5.0f KiB (%s)\n", "Total", total / 1024., total;
-  }
-}