diff --git a/docs/requirements.txt b/docs/requirements.txt
index b1bd611df0..d155c265ad 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -40,6 +40,6 @@ onnx>=1.13.0
 onnxruntime; python_version <= '3.10'
 zarr
 huggingface_hub
-pyamg>=5.0.0
+pyamg>=5.0.0, <5.3.0
 packaging
 polygraphy
diff --git a/requirements-dev.txt b/requirements-dev.txt
index 87840556ee..fff622b021 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -59,7 +59,7 @@ zarr
 lpips==0.1.4
 nvidia-ml-py
 huggingface_hub
-pyamg>=5.0.0
+pyamg>=5.0.0, <5.3.0
 git+https://github.com/facebookresearch/segment-anything.git@6fdee8f2727f4506cfbbe553e23b895e27956588
 onnx_graphsurgeon
 polygraphy
diff --git a/setup.cfg b/setup.cfg
index fc415e6cc0..b3949213c2 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -87,7 +87,7 @@ all =
     lpips==0.1.4
     nvidia-ml-py
     huggingface_hub
-    pyamg>=5.0.0
+    pyamg>=5.0.0, <5.3.0
 nibabel =
     nibabel
 ninja =
@@ -175,7 +175,7 @@ polygraphy =
 huggingface_hub =
     huggingface_hub
 pyamg =
-    pyamg>=5.0.0
+    pyamg>=5.0.0, <5.3.0
 # segment-anything =
 #     segment-anything @ git+https://github.com/facebookresearch/segment-anything@6fdee8f2727f4506cfbbe553e23b895e27956588#egg=segment-anything