diff --git a/pyproject.toml b/pyproject.toml
index c5f17e6..97a990e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -74,6 +74,7 @@ zarr = "*"
 tensorly = "*"
 torchvision = "*"
 torchaudio = "*"
+sparse = "*"
 torchtext = "*"
 triton = {version = "2.2.0", optional = true}
 pytorch-fast-transformers = { version = "*", optional = true }