[TRTLLM-11851][feat] MX and GMS integration MVP for Dynamo weight sharing

setup.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2022-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2022-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -427,6 +427,32 @@ def extract_from_precompiled(precompiled_location: str, package_data: List[str],
     scripts=['tensorrt_llm/llmapi/trtllm-llmapi-launch'],
     extras_require={
         "devel": devel_deps,
+        # NOTE: The MX (modelexpress) and GMS (gpu-memory-service) Python
+        # packages used by tensorrt_llm._torch.models.checkpoints.mx and
+        # tensorrt_llm._torch.memory.gpu_memory_backend are intentionally
+        # NOT declared as ``[mx]`` / ``[gms]`` / ``[dynamo]`` extras here
+        # while the integration is at prototype status. Two reasons:
+        #
+        #   1. ``gpu-memory-service`` v0.9.0 ships only as source under
+        #      ``ai-dynamo/dynamo/lib/gpu_memory_service/`` and is not
+        #      published to PyPI, so NVIDIA's OSS vulnerability scanner
+        #      cannot resolve a pinned version on it.
+        #   2. ``modelexpress`` v0.3.0 is on PyPI (Apache-2.0) but is
+        #      brand-new (Beta status, single release) and still needs
+        #      onboarding into NVIDIA's OSS package allowlist.
+        #
+        # Until both packages are PyPI-published *and* allowlisted, users
+        # who want to exercise the MX or GMS code paths install the deps
+        # manually:
+        #
+        #     pip install "modelexpress>=0.3.0,<0.4.0"
+        #     git clone https://github.com/ai-dynamo/dynamo \
+        #       && pip install ./dynamo/lib/gpu_memory_service
+        #
+        # Restoring one-line ``pip install tensorrt_llm[dynamo]``
+        # ergonomics is a single revert of this hunk once the upstream
+        # publish + OSS-allowlist steps are complete (tracked in §15 of
+        # the design doc as MX-7 and GMS-6).
     },
     zip_safe=True,
     install_requires=required_deps,

tensorrt_llm/_torch/memory/__init__.py

@@ -0,0 +1,18 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# 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
+#
+# http://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.
+
+from .gpu_memory_backend import GMSBackend, GPUMemoryBackend
+
+__all__ = ["GPUMemoryBackend", "GMSBackend"]