diff --git a/.gitignore b/.gitignore
index d4189547..818b4b76 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,6 +4,7 @@ __pycache__
 *.egg-info
 build
 outputs
+dist/*
 
 # data
 data
diff --git a/pyproject.toml b/pyproject.toml
index 55007a7b..5fc4a7b2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,48 @@
-[tool.pydoclint]
-style = 'google'
-check-return-types = 'False'
+# ---- All project specifications ---- #
+[project]
+name = "torchtitan"
+version = "0.0.2"
+description = "A native-PyTorch library for large scale LLM training"
+readme = "README.md"
+requires-python = ">=3.8"
+license = {file = "LICENSE"}
+authors = [
+    { name = "PyTorch Team", email = "packages@pytorch.org" },
+]
+keywords = ["pytorch", "training", "llm"]
+dependencies = [
+    # Hugging Face integrations
+    "datasets",
+
+    # Miscellaneous
+    "sentencepiece",
+    "tomli >= 1.1.0"
+]
+
+[project.urls]
+GitHub = "https://github.com/pytorch/torchtitan"
+Documentation = "https://github.com/pytorch/torchtitan/tree/main/docs"
+Issues = "https://github.com/pytorch/torchtitan/issues"
+
+[project.optional-dependencies]
+dev = [
+    "pre-commit",
+    "pytest",
+    "pytest-cov",
+    "tensorboard",
+]
+
+# ---- Explicit project build information ---- #
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = [""]
+include = ["torchtrain*"]
+
+[tool.setuptools.package-data]
+recipes = ["train_configs/*.toml"]
 
 [tool.pytest.ini_options]
 addopts = ["--showlocals"] # show local variables in tracebacks
diff --git a/setup.py b/setup.py
index 21269cdb..f5d01e3c 100644
--- a/setup.py
+++ b/setup.py
@@ -16,5 +16,5 @@
     description="Package for training large models using native PyTorch",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",
-    url="https://github.com/pytorch-labs/torchtitan",
+    url="https://github.com/pytorch/torchtitan",
 )
diff --git a/version.txt b/version.txt
new file mode 100644
index 00000000..4e379d2b
--- /dev/null
+++ b/version.txt
@@ -0,0 +1 @@
+0.0.2