diff --git a/setup.py b/setup.py
index 511d394..cf14b63 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import find_packages, setup
 
 install_requires = [
-    'torch>=2.3.0,<2.4',
+    'torch>=2.3.0,<3.0',
     'triton>=2.1.0',
 ]
 
@@ -15,7 +15,7 @@
 
 setup(
     name="stanford-stk",
-    version="0.7.0",
+    version="0.7.1",
     author="Trevor Gale",
     author_email="tgale@stanford.edu",
     description="Sparse Toolkit",