diff --git a/fused_ssim/__init__.py b/fused_ssim/__init__.py
index 776fed7..365ae03 100644
--- a/fused_ssim/__init__.py
+++ b/fused_ssim/__init__.py
@@ -1,5 +1,3 @@
-from typing import NamedTuple
-import torch.nn as nn
 import torch
 from fused_ssim_cuda import fusedssim, fusedssim_backward
 
diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..01bbc3b
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,27 @@
+[build-system]
+requires = [
+    "setuptools>=42",
+    "wheel",
+    "torch>=2.6.0",
+    "ninja",  # Required for CUDA extensions
+    "typing-extensions>=4.0.0",  # Required by torch.utils.cpp_extension
+]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "fused_ssim"
+version = "0.1.2"
+description = "CUDA-accelerated SSIM implementation"
+authors = [
+    {name = "Rahul Goel", email = "rahulgoellko@gmail.com"},
+]
+requires-python = ">=3.9"
+dependencies = [
+    "torch>=2.6.0",
+]
+
+[project.urls]
+repository = "https://github.com/yourusername/fused-ssim"
+
+[tool.setuptools]
+packages = ["fused_ssim"]