diff --git a/poetry.lock b/poetry.lock
index 0fb5312..3e93fa3 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1902,6 +1902,22 @@ files = [
 dev = ["pre-commit", "tox"]
 testing = ["coverage", "pytest", "pytest-benchmark"]
 
+[[package]]
+name = "polygraphy"
+version = "0.47.1"
+description = "Polygraphy: A Deep Learning Inference Prototyping and Debugging Toolkit"
+optional = false
+python-versions = ">=3.6"
+groups = ["main"]
+files = [
+    {file = "polygraphy-0.47.1-py2.py3-none-any.whl", hash = "sha256:6e01839686de6852a6544f147e82be8f37f678ec929f31eb0bf6747ffe4a2646"},
+]
+
+[package.source]
+type = "legacy"
+url = "https://pypi.ngc.nvidia.com"
+reference = "ngc"
+
 [[package]]
 name = "pre-commit"
 version = "4.3.0"
@@ -3340,4 +3356,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10,<3.15"
-content-hash = "344541e8a3c431274a7375537bb08d656b7f43ab2f29238254f8a4de831db7a1"
+content-hash = "26533f917997bf4b86912c0ab5c81a77c318c0b7868aa451dde59a606122d43a"
diff --git a/pyproject.toml b/pyproject.toml
index 62b1012..6eef370 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,6 +21,7 @@ dependencies = [
     "xformers (>=0.0.29.post3,<0.0.30)",
     "streamdiffusion[tensorrt] @ git+https://github.com/cumulo-autumn/StreamDiffusion.git@b623251dc055e1fd858d53509aa43e09dfc5cdc0",
     "tensorrt (>=10.14.1.48.post1,<11.0.0.0)",
+    "polygraphy (>=0.47.1,<0.48.0)",
 ]
 
 [tool.poetry]
@@ -47,6 +48,7 @@ torchaudio = { version = ">=2.6.0,<3.0.0", source = "pytorch-cu124" }
 torchvision = { version = ">=0.21.0,<0.25.0", source = "pytorch-cu124" }
 xformers = { version = ">=0.0.29.post3,<0.0.30", source = "pytorch-cu124" }
 tensorrt = { version = ">=10.14.1.48.post1,<11.0.0.0", source = "nvidia" }
+polygraphy = { version = ">=0.47.1,<0.48.0", source = "ngc" }
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
diff --git a/requirements.txt b/requirements.txt
index 82bd808..8eb301b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 --extra-index-url https://pypi.nvidia.com
+--extra-index-url https://pypi.ngc.nvidia.com
 --extra-index-url https://download.pytorch.org/whl/cu124
 
 accelerate==1.11.0 ; python_version >= "3.10" and python_version < "3.15"
@@ -47,6 +48,7 @@ onnx==1.15.0 ; python_version >= "3.10" and python_version < "3.15"
 onnxruntime==1.16.3 ; python_version >= "3.10" and python_version < "3.15"
 packaging==25.0 ; python_version >= "3.10" and python_version < "3.15"
 pillow==12.0.0 ; python_version >= "3.10" and python_version < "3.15"
+polygraphy==0.47.1 ; python_version >= "3.10" and python_version < "3.15"
 protobuf==3.20.2 ; python_version >= "3.10" and python_version < "3.15"
 psutil==7.1.2 ; python_version >= "3.10" and python_version < "3.15"
 pyreadline3==3.5.4 ; python_version >= "3.10" and python_version < "3.15" and sys_platform == "win32"