upgrade monarch & torchstore version (#657)

Author: JenniferWang

assets/versions.sh

@@ -10,9 +10,9 @@
 # Stable versions of upstream libraries for OSS repo
 PYTORCH_VERSION="2.9.0"
 VLLM_VERSION="v0.10.0"
-MONARCH_VERSION="0.1.2"
+MONARCH_NIGHTLY_VERSION="2025.12.17"
 TORCHTITAN_VERSION="0.2.0"
-TORCHSTORE_VERSION="0.1.2"
+TORCHSTORE_BRANCH="no-monarch-2025.12.17"
 
 # Torchtitan commit hash for launching on MAST
 TORCHTITAN_COMMIT_MAST="d0e25450bcac2332359b13fbda430dc701f073d4"

pyproject.toml

@@ -14,8 +14,9 @@ dependencies = [
     "torch==2.9.0",
     "torchdata>=0.8.0",
     "torchtitan==0.2.0",
-    "torchmonarch==0.1.2",
-    "torchstore==0.1.2",
+    "torchmonarch-nightly==2025.12.17",
+    # Issue 656: switch to ping torchstore nightly
+    "torchstore",
     # vLLM
     "vllm",
     # Hugging Face integrations
@@ -47,7 +48,6 @@ dev = [
     "anyio",
     "pytest-asyncio",
     "multiprocess",
-    "langid",
 ]
 docs = [
     "sphinx==7.2.6",
@@ -56,6 +56,7 @@ docs = [
     "sphinx-design==0.6.1",
     "sphinxcontrib-mermaid==1.0.0",
     "sphinx-gallery==0.19.0",
+    "matplotlib",
     "myst-parser",
     "sphinx-sitemap==2.7.1",
     "sphinx-autodoc-typehints==1.25.3",
@@ -92,6 +93,8 @@ url = "https://download.pytorch.org/whl/preview/forge"
 [tool.uv.sources]
 torch = { index = "pytorch-cu128" }
 vllm = { index = "vllm-forge" }
+# Issue 656: switch to ping torchstore nightly
+torchstore = { git = "https://github.com/meta-pytorch/torchstore.git", branch = "no-monarch-2025.12.17" }
 
 [tool.uv]
 # TODO: revert to stricter default uv strategy

scripts/install.sh

@@ -198,12 +198,14 @@ main() {
     pip install "setuptools<80"
     python -m pip install vllm --no-cache-dir --index-url https://download.pytorch.org/whl/preview/forge
 
-    # Install monarch
-    pip install torchmonarch==$MONARCH_VERSION
+    # Install monarch nightly
+    pip install torchmonarch-nightly==$MONARCH_NIGHTLY_VERSION
 
-    # Install torchtitan and torchstore
+    # Install torchtitan
     pip install torchtitan==$TORCHTITAN_VERSION
-    pip install torchstore==$TORCHSTORE_VERSION
+
+    # Install torchstore without monarch or torch dependency
+    pip install "git+https://github.com/meta-pytorch/torchstore.git@${TORCHSTORE_BRANCH}"
 
     log_info "Installing Forge from source..."
     pip install -e ".[dev]"