diff --git a/examples/MODULE.bazel.lock b/examples/MODULE.bazel.lock
index c2861d8..aeb27f0 100644
--- a/examples/MODULE.bazel.lock
+++ b/examples/MODULE.bazel.lock
@@ -1265,7 +1265,7 @@
           "@@rules_python+//tools/publish/requirements_darwin.txt": "6d10732737e5504291360b78fbf2b82abe39a5a021bd937f987f8433519b0f2c",
           "@@rules_python+//tools/publish/requirements_linux.txt": "cb63e4751377d1722a81fd5835b0526208620f1303feb473ed0dd45ae4d71ba9",
           "@@rules_python+//tools/publish/requirements_windows.txt": "f92da1a252df04e1931a0e16dfbc5a3c165a85b796b7fc5e56ee4e3846d82558",
-          "@@zml+//runtimes/neuron/requirements.lock.txt": "a5e5353b149b33dc067ce27ef67896d8ca3518e37a344e215e6604f98ea6d0fc"
+          "@@zml+//runtimes/neuron/requirements.lock.txt": "fa0b5b967162e303415ee1dd7d2c1a448068935663ba5ac5b5f72a515fe3f6f6"
         },
         "recordedDirentsInputs": {},
         "envVariables": {
@@ -1296,7 +1296,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "boto3==1.38.25 --hash=sha256:2f2cd517dd31d33ace0eefe567dc903fdf74221513e32f1e9445bdfac7554db7 --hash=sha256:85c1556a110896f68de8573a9b4757c81071448dbf6ffc1074941bfc8a43195e"
+              "requirement": "boto3==1.39.9 --hash=sha256:5bc85e9fdec4e21ef5ca2c22b4d51a3e32b53f3da36ce51f5a3ea4dbde07b132 --hash=sha256:e3d3a6b617e1575e7ec854c820a882ab2e189a0421e74dc0dca2c9e13d4370a5"
             }
           },
           "neuron_py_deps_311_botocore": {
@@ -1322,7 +1322,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "botocore==1.38.25 --hash=sha256:5a960bd990a11cdb78865e908a58ed712d87d9b419f55ad21c99d7d21f0d6726 --hash=sha256:8c73e97d9662a6c92be33dab66cd1e2b59797154c7ec379ce3bb5d6779d9d78c"
+              "requirement": "botocore==1.39.9 --hash=sha256:02f141c2849e4589a79feea245ce4ecc478d48b7865572445af8aae3b041772d --hash=sha256:a9691cbe03a3bc8b2720b3c36e5c5a2eecace6acd72bfb1107f00e75edaec4f3"
             }
           },
           "neuron_py_deps_311_certifi": {
@@ -1348,7 +1348,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "certifi==2025.4.26 --hash=sha256:0a816057ea3cdefcef70270d2c515e4506bbc954f417fa5ade2021213bb8f0c6 --hash=sha256:30350364dfe371162649852c63336a15c70c6510c2ad5015b21c2345311805f3"
+              "requirement": "certifi==2025.7.14 --hash=sha256:6b31f564a415d79ee77df69d757bb49a5bb53bd9f756cbbe24394ffd6fc1f4b2 --hash=sha256:8ea99dbdfaaf2ba2f9bac77b9249ef62ec5218e7c2b2e903378ed5fccf765995"
             }
           },
           "neuron_py_deps_311_charset_normalizer": {
@@ -1504,7 +1504,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "libneuronxla==2.2.3493.0+78c3e78c --hash=sha256:9ba54e5d481afc25ae83e964a7c076543991ad80b88839cd500abcac130595e8"
+              "requirement": "libneuronxla==2.2.4410.0+835a67fb --hash=sha256:37e5f08482ef1a9a844c2894de6d91c03400a092764e54ab67612456be439f16"
             }
           },
           "neuron_py_deps_311_lockfile": {
@@ -1608,7 +1608,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "neuronx-cc==2.18.121.0+9e31e41a --hash=sha256:b4c0959594af716aff94698e15cc3997cc39a66c853c25a1bbf5eb7bf71fd7ec"
+              "requirement": "neuronx-cc==2.19.8089.0+8ab9f450 --hash=sha256:fbb05d8a0da8f54954c03ff2bea659f627309110c00efd54f523d877ddd6b88d"
             }
           },
           "neuron_py_deps_311_numpy": {
@@ -1790,7 +1790,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "requests==2.31.0 --hash=sha256:58cd2187c01e70e6e26505bca751777aa9f2ee0b7f4300988b709f44e013003f --hash=sha256:942c5a758f98d790eaed1a29cb6eefc7ffb0d1cf7af05c3d2791656dbd6ad1e1"
+              "requirement": "requests==2.32.4 --hash=sha256:27babd3cda2a6d50b30443204ee89830707d396671944c998b5975b031ac2b2c --hash=sha256:27d0316682c8a29834d3264820024b62a36942083d52caf2f14c0591336d3422"
             }
           },
           "neuron_py_deps_311_requests_unixsocket": {
@@ -1842,7 +1842,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "s3transfer==0.13.0 --hash=sha256:0148ef34d6dd964d0d8cf4311b2b21c474693e57c2e069ec708ce043d2b527be --hash=sha256:f5e6db74eb7776a37208001113ea7aa97695368242b364d73e91c981ac522177"
+              "requirement": "s3transfer==0.13.1 --hash=sha256:a981aa7429be23fe6dfc13e80e4020057cbab622b08c0315288758d67cabc724 --hash=sha256:c3fdba22ba1bd367922f27ec8032d6a1cf5f10c934fb5d68cf60fd5a23d936cf"
             }
           },
           "neuron_py_deps_311_scipy": {
@@ -1923,6 +1923,32 @@
               "requirement": "tqdm==4.67.1 --hash=sha256:26445eca388f82e72884e0d580d5464cd801a3ea01e63e5601bdff9ba6a48de2 --hash=sha256:f8aef9c52c08c13a65f30ea34f4e5aac3fd1a34959879d7e59e63027286627f2"
             }
           },
+          "neuron_py_deps_311_typing_extensions": {
+            "repoRuleId": "@@rules_python+//python/private/pypi:whl_library.bzl%whl_library",
+            "attributes": {
+              "dep_template": "@neuron_py_deps//{name}:{target}",
+              "download_only": true,
+              "experimental_target_platforms": [
+                "cp311_linux_x86_64"
+              ],
+              "extra_pip_args": [
+                "--index-url",
+                "https://pypi.org/simple",
+                "--extra-index-url",
+                "https://pip.repos.neuron.amazonaws.com",
+                "--find-links",
+                "https://mirror.zml.ai/pypi/aws-neuronx-runtime-discovery/index.html",
+                "--abi=cp311",
+                "--implementation=cp",
+                "--python-version=311",
+                "--platform=linux_x86_64",
+                "--platform=manylinux2014_x86_64"
+              ],
+              "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
+              "repo": "neuron_py_deps_311",
+              "requirement": "typing-extensions==4.14.1 --hash=sha256:38b39f4aeeab64884ce9f74c94263ef78f3c22467c8724005483154c26648d36 --hash=sha256:d1e1e3b58374dc93031d6eda2420a48ea44a36c2b4766a4fdeb3710755731d76"
+            }
+          },
           "neuron_py_deps_311_urllib3": {
             "repoRuleId": "@@rules_python+//python/private/pypi:whl_library.bzl%whl_library",
             "attributes": {
@@ -1946,7 +1972,7 @@
               ],
               "python_interpreter_target": "@@rules_python++python+python_3_11_host//:python",
               "repo": "neuron_py_deps_311",
-              "requirement": "urllib3==2.4.0 --hash=sha256:414bc6535b787febd7567804cc015fee39daab8ad86268f1310a9250697de466 --hash=sha256:4e16665048960a0900c702d4a66415956a584919c03361cac9f1df5c5dd7e813"
+              "requirement": "urllib3==2.5.0 --hash=sha256:3fc47733c7e419d4bc3f6b3dc2b4f890bb743906a30d56ba4a5bfa4bbff92760 --hash=sha256:e6b01673c0fa6a13e374b50871808eb3bf7046c4b125b216f6bf1cc604cff0dc"
             }
           },
           "pip_deps_310_numpy": {
@@ -4375,6 +4401,7 @@
                 "scipy": "{\"neuron_py_deps_311_scipy\":[{\"version\":\"3.11\"}]}",
                 "six": "{\"neuron_py_deps_311_six\":[{\"version\":\"3.11\"}]}",
                 "tqdm": "{\"neuron_py_deps_311_tqdm\":[{\"version\":\"3.11\"}]}",
+                "typing_extensions": "{\"neuron_py_deps_311_typing_extensions\":[{\"version\":\"3.11\"}]}",
                 "urllib3": "{\"neuron_py_deps_311_urllib3\":[{\"version\":\"3.11\"}]}"
               },
               "packages": [
@@ -4403,6 +4430,7 @@
                 "scipy",
                 "six",
                 "tqdm",
+                "typing_extensions",
                 "urllib3"
               ],
               "groups": {}