diff --git a/pyproject.toml b/pyproject.toml
index df57acf10..1af32755f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
     authors=[{name="Neel Nanda", email="77788841+TransformerLensOrg@users.noreply.github.com"}]
     dependencies=[
-        "accelerate>=0.23.0",                  # Needed for Llama Models
+        "accelerate>=0.23.0",                                                # Needed for Llama Models
         "beartype>=0.14.1",
         "better-abc>=0.0.3",
         "datasets>=2.7.1",
@@ -26,6 +26,7 @@
     description="An implementation of transformers tailored for mechanistic interpretability."
     license={text="MIT"}
     name="transformer-lens"
+    packages=[{include="transformer_lens"}, {include="transformer_lens/py.typed"}]
     readme="README.md"
     requires-python=">=3.10,<4.0"
     version="0.0.0"
@@ -85,13 +86,11 @@
             "-W ignore::beartype.roar.BeartypeDecorHintPep585DeprecationWarning",
         ]
         doctest_optionflags="NORMALIZE_WHITESPACE ELLIPSIS FLOAT_CMP"
-        markers=[
-            "slow: marks tests as slow (deselect with '-m \"not slow\"')",
-        ]
         filterwarnings=[
-            "ignore:pkg_resources is deprecated as an API:DeprecationWarning",
             "ignore:distutils Version classes are deprecated:DeprecationWarning",
+            "ignore:pkg_resources is deprecated as an API:DeprecationWarning",
         ]
+        markers=["slow: marks tests as slow (deselect with '-m \"not slow\"')"]
         pythonpath=["."]
         testpaths=["tests", "transformer_lens"] # Only test these directories
 
diff --git a/transformer_lens/py.typed b/transformer_lens/py.typed
new file mode 100644
index 000000000..e69de29bb