diff --git a/tpu_sync/frameworks/jax/BUILD b/tpu_sync/frameworks/jax/BUILD
index 8d44e43d..f9785747 100644
--- a/tpu_sync/frameworks/jax/BUILD
+++ b/tpu_sync/frameworks/jax/BUILD
@@ -144,6 +144,7 @@ nanobind_extension(
         "//tpu_sync/core:raw_transfer_core",
         "//tpu_sync/kv_cache:kv_cache_store",
         "//tpu_sync/kv_cache:kv_cache_store_wrapper",
+        "//tpu_sync/telemetry/python:telemetry_binding",
         "//tpu_sync/weight_sync:weight_synchronizer_base",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
diff --git a/tpu_sync/frameworks/jax/tpu_raiden_jax_module.cc b/tpu_sync/frameworks/jax/tpu_raiden_jax_module.cc
index e07408a0..70065097 100644
--- a/tpu_sync/frameworks/jax/tpu_raiden_jax_module.cc
+++ b/tpu_sync/frameworks/jax/tpu_raiden_jax_module.cc
@@ -42,6 +42,7 @@
 #include "tpu_sync/frameworks/jax/weight_synchronizer.h"
 #include "tpu_sync/kv_cache/kv_cache_store.h"
 #include "tpu_sync/kv_cache/kv_cache_store_wrapper.h"
+#include "tpu_sync/telemetry/python/telemetry_binding.h"
 
 namespace nb = nanobind;
 
@@ -784,4 +785,6 @@ NB_MODULE(_tpu_raiden_jax, m) {
              return std::make_tuple(py_done, py_failed, py_pending, py_existing,
                                     py_unregistered);
            });
+
+  ::tpu_raiden::telemetry::BindTelemetryApi(m);
 }
diff --git a/tpu_sync/frameworks/torch/BUILD b/tpu_sync/frameworks/torch/BUILD
index 36d6d6fe..70c0b2a4 100644
--- a/tpu_sync/frameworks/torch/BUILD
+++ b/tpu_sync/frameworks/torch/BUILD
@@ -207,6 +207,7 @@ nanobind_extension(
         "//tpu_sync/kv_cache:kv_cache_store_wrapper",
         "//tpu_sync/kv_cache/reshard:reshard_client",
         "//tpu_sync/rpc:raiden_service_cc_proto",
+        "//tpu_sync/telemetry/python:telemetry_binding",
         "//tpu_sync/weight_sync:weight_synchronizer_base",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
diff --git a/tpu_sync/frameworks/torch/tpu_raiden_torch_module.cc b/tpu_sync/frameworks/torch/tpu_raiden_torch_module.cc
index f98182b6..6fe71d0e 100644
--- a/tpu_sync/frameworks/torch/tpu_raiden_torch_module.cc
+++ b/tpu_sync/frameworks/torch/tpu_raiden_torch_module.cc
@@ -43,6 +43,7 @@
 #include "tpu_sync/kv_cache/kv_cache_store_wrapper.h"
 #include "tpu_sync/kv_cache/reshard/reshard_client.h"
 #include "tpu_sync/rpc/raiden_service.pb.h"
+#include "tpu_sync/telemetry/python/telemetry_binding.h"
 
 namespace nb = nanobind;
 
@@ -1110,4 +1111,6 @@ NB_MODULE(_tpu_raiden_torch, m) {
       nb::arg("raiden_controller_port") = 0,
       nb::arg("reshard_service_port") = 0,
       nb::call_guard<nb::gil_scoped_release>());
+
+  tpu_raiden::telemetry::BindTelemetryApi(m);
 }