diff --git a/setup.py b/setup.py
index 26cecf4c..1b80f17c 100644
--- a/setup.py
+++ b/setup.py
@@ -70,7 +70,7 @@
 extra_deps = {}
 
 extra_deps['gg'] = [
-    'grouped_gemm==0.1.6',
+    'grouped_gemm==0.2.0',
 ]
 
 extra_deps['dev'] = [