diff --git a/.github/workflows/mypy.yml b/.github/workflows/mypy.yml
index 6081a268..b9e6c210 100644
--- a/.github/workflows/mypy.yml
+++ b/.github/workflows/mypy.yml
@@ -50,7 +50,7 @@ jobs:
         pip install -I ibm-fms@git+https://github.com/foundation-model-stack/foundation-model-stack@main
 
         # No type stubs available for "fire" and "transformers"
-        mypy --exclude fms_to_hf.py --exclude main_training.py .
+        mypy --exclude fms_to_hf.py --exclude main_training.py --exclude setup.py .
 
     - name: Save Virtualenv
       id: cache-venv-save
diff --git a/setup.py b/setup.py
new file mode 100644
index 00000000..d93301ab
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,18 @@
+from setuptools import find_packages, setup
+
+
+setup(
+    name="fms_fsdp",
+    version="0.0.1",
+    author="Linsong Chu, Davis Wertheimer, Brian Vaughan, Andrea Frittoli, Joshua Rosenkranz, Antoni Viros i Martin, Raghu Kiran Ganti",
+    author_email="lchu@us.ibm.com",
+    description="Pretraining scripts using FSDP and IBM Foundation Model Stack",
+    url="https://github.com/foundation-model-stack/fms-fsdp",
+    packages=find_packages(),
+    install_requires=["ibm-fms >= 0.0.3", "torch >= 2.1"],
+    license="Apache License 2.0",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: Apache Software License",
+    ],
+)