diff --git a/DOC_SITE_SETUP.md b/DOC_SITE_SETUP.md
index 8b0313d..b8779f0 100644
--- a/DOC_SITE_SETUP.md
+++ b/DOC_SITE_SETUP.md
@@ -16,7 +16,7 @@ This repository is configured to build documentation locally and publish the gen
    - Branch: `main`
    - Folder: `/ (root)`
 4. Create a token that has write access to `INFTY-AI/doc`.
-5. Add that token as the `DOC_REPO_TOKEN` secret in the source repository `WanNaa/INFTY_demo`.
+5. Add that token as the `DOC_REPO_TOKEN` secret in the source repository `THUDM/INFTY`.
 
 ## Bootstrap the target repository locally
 
diff --git a/README.md b/README.md
index fabd3ac..5118fbb 100644
--- a/README.md
+++ b/README.md
@@ -36,7 +36,7 @@
 # 🌈 What is INFTY?
 
 <div align="center">
-<img src="./img/INFTY_demo.gif" alt="animated" />
+<img src="./img/INFTY.gif" alt="animated" />
 </div>
 </br>
 
diff --git a/bootstrap_doc_site_repo.sh b/bootstrap_doc_site_repo.sh
index dca1692..2b4156f 100755
--- a/bootstrap_doc_site_repo.sh
+++ b/bootstrap_doc_site_repo.sh
@@ -131,7 +131,7 @@ cat > "$TARGET_DIR/index.html" <<'EOF'
       <h1>INFTY documentation is being prepared.</h1>
       <p>
         This repository is intended to host the published static site for
-        <a href="https://github.com/WanNaa/INFTY_demo">WanNaa/INFTY_demo</a>.
+        <a href="https://github.com/THUDM/INFTY">THUDM/INFTY</a>.
       </p>
       <p>
         Once the source repository finishes its documentation workflow, this
@@ -164,6 +164,6 @@ Next steps:
   1. Create the GitHub repository INFTY-AI/doc if it does not exist yet.
   2. Commit and push the files in $TARGET_DIR to the main branch of that repository.
   3. Enable GitHub Pages for the main branch root in INFTY-AI/doc.
-  4. Add a DOC_REPO_TOKEN secret to WanNaa/INFTY_demo with write access to INFTY-AI/doc.
+  4. Add a DOC_REPO_TOKEN secret to THUDM/INFTY with write access to INFTY-AI/doc.
   5. Trigger the Documentation workflow in this source repository.
 EOF
diff --git a/docs/citation.md b/docs/citation.md
index a777633..b9e2907 100644
--- a/docs/citation.md
+++ b/docs/citation.md
@@ -11,7 +11,7 @@ For the toolkit, use a software citation similar to:
   title        = {INFTY: An Optimization Toolkit to Support Continual AI},
   author       = {INFTY contributors},
   year         = {2026},
-  url          = {https://github.com/WanNaa/INFTY_demo}
+  url          = {https://github.com/THUDM/INFTY}
 }
 ```
 
diff --git a/docs/developer_guide.md b/docs/developer_guide.md
index a495a1e..519f7f5 100644
--- a/docs/developer_guide.md
+++ b/docs/developer_guide.md
@@ -7,7 +7,7 @@ This guide describes how to extend INFTY with new optimizers, plotting utilities
 A typical INFTY repository layout is:
 
 ```text
-INFTY_demo/
+INFTY/
 ├── README.md
 ├── pyproject.toml
 ├── setup.py
@@ -32,8 +32,8 @@ INFTY_demo/
 ## Development installation
 
 ```bash
-git clone https://github.com/WanNaa/INFTY_demo.git
-cd INFTY_demo
+git clone https://github.com/THUDM/INFTY.git
+cd INFTY
 python -m venv .venv
 source .venv/bin/activate
 python -m pip install --upgrade pip
diff --git a/docs/examples.md b/docs/examples.md
index 641a5d9..eac2936 100644
--- a/docs/examples.md
+++ b/docs/examples.md
@@ -6,12 +6,12 @@ For the minimum usage pattern and the smallest inline optimizer example, start w
 
 ## PILOT formal runs
 
-The repository retains formal [PILOT](https://github.com/WanNaa/INFTY_demo/tree/main/workdirs/PILOT) launcher scripts under `workdirs/scripts/`. These correspond to the preserved formal artifacts under `workdirs/results/`.
+The repository retains formal [PILOT](https://github.com/THUDM/INFTY/tree/main/workdirs/PILOT) launcher scripts under `workdirs/scripts/`. These correspond to the preserved formal artifacts under `workdirs/results/`.
 
 Install the PILOT/demo dependencies from the repository root:
 
 ```bash
-cd INFTY_demo
+cd INFTY
 python -m pip install ".[examples]"
 ```
 
diff --git a/docs/installation.md b/docs/installation.md
index 2680411..9e89f52 100644
--- a/docs/installation.md
+++ b/docs/installation.md
@@ -32,8 +32,8 @@ python -c "from infty.optim import C_Flat, ZeroFlow, UniGrad_FS; print('INFTY im
 ## Install from source
 
 ```bash
-git clone https://github.com/WanNaa/INFTY_demo.git
-cd INFTY_demo
+git clone https://github.com/THUDM/INFTY.git
+cd INFTY
 python -m pip install --upgrade pip
 python -m pip install .
 ```
diff --git a/img/INFTY_demo.gif b/img/INFTY.gif
similarity index 100%
rename from img/INFTY_demo.gif
rename to img/INFTY.gif
diff --git a/mkdocs.yml b/mkdocs.yml
index 6241ece..36d49c6 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -1,8 +1,8 @@
 site_name: INFTY
 site_description: Optimization-centric toolkit for Continual AI
 site_url: https://INFTY-AI.github.io/doc/
-repo_url: https://github.com/WanNaa/INFTY_demo
-repo_name: WanNaa/INFTY_demo
+repo_url: https://github.com/THUDM/INFTY
+repo_name: THUDM/INFTY
 edit_uri: edit/main/docs/
 
 extra:
diff --git a/pyproject.toml b/pyproject.toml
index 664d4ab..53f7651 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -37,10 +37,10 @@ dependencies = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/WanNaa/INFTY_demo"
+Homepage = "https://github.com/THUDM/INFTY"
 Documentation = "https://infty-ai.github.io/doc/"
-Repository = "https://github.com/WanNaa/INFTY_demo"
-Issues = "https://github.com/WanNaa/INFTY_demo/issues"
+Repository = "https://github.com/THUDM/INFTY"
+Issues = "https://github.com/THUDM/INFTY/issues"
 
 [project.optional-dependencies]
 examples = [
diff --git a/setup.py b/setup.py
index d26eab6..995bfa1 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,7 @@ def read_readme():
     long_description_content_type="text/markdown",
     author='taofeng weili',
     author_email='ymjiii98@gmail.com',  
-    url='https://github.com/WanNaa/INFTY_demo',
+    url='https://github.com/THUDM/INFTY',
     packages=find_packages(where='src'),
     package_dir={'': 'src'},
     classifiers=[