Add setup.py

setup.py

@@ -0,0 +1,26 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="sensitive-attribute-circuit-analysis",
+    version="0.1.0",
+    description="Mechanistic interpretability study: Do LLMs route through sensitive-attribute circuits that shift analytical choices?",
+    author="ryanyen22",
+    packages=find_packages(),
+    python_requires=">=3.9",
+    install_requires=[
+        "torch>=2.0.0",
+        "transformers>=4.40.0",
+        "accelerate>=0.27.0",
+        "bitsandbytes>=0.43.0",
+        "scikit-learn>=1.3.0",
+        "numpy>=1.24.0",
+        "matplotlib>=3.7.0",
+        "huggingface_hub>=0.20.0",
+    ],
+    extras_require={
+        "nnsight": ["nnsight>=0.3.0"],
+        "pyvene": ["pyvene>=0.1.0"],
+        "viz": ["circuitsvis>=1.0.0", "plotly>=5.0.0"],
+        "all": ["nnsight>=0.3.0", "pyvene>=0.1.0", "circuitsvis>=1.0.0"],
+    },
+)