Made oss dependencies optional (#47)

* Made oss dependencies optional

* Remove extra comments

Author: pbontrager

README.md

@@ -31,13 +31,7 @@ uv sync --all-extras
 source .venv/bin/activate
 ```
 
-2. Install Monarch
-
-```bash
-uv pip install torchmonarch-nightly==2025.8.1
-```
-
-3. Setup CUDA on local machine
+2. Setup CUDA on local machine
 
 ```bash
 # feature install if you don't have /user/local/cuda-12.8
@@ -54,7 +48,7 @@ export CUDA_CUDART_LIBRARY=$CUDA_HOME/lib64/libcudart.so
 export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
 ```
 
-4 Build vllm from source
+3. Build vllm from source
 
 ```bash
 git clone https://github.com/vllm-project/vllm.git --branch v0.10.0

pyproject.toml

@@ -13,10 +13,6 @@ dependencies = [
     # PyTorch
     "torchdata>=0.8.0",
     "torchtitan",
-    # Temporarily comment out in favor of enabling MAST path
-    #"torch",
-    # TODO: uncomment when this doesn't break MAST
-    #"torchmonarch-nightly==2025.8.1",
     # vLLM
     # TODO: pin specific vllm version
     #"vllm==0.10.0",
@@ -43,6 +39,12 @@ dev = [
     "anyio",
     "pytest-asyncio",
 ]
+oss = [
+    "torch",
+    "torchmonarch-nightly==2025.8.1",
+    "torchstore",
+]
+
 
 [project.scripts]
 forge = "forge.cli.forge:main"
@@ -85,6 +87,7 @@ url  = "https://download.pytorch.org/whl/nightly/cu128"
 [tool.uv.sources]
 torchtitan = { index = "pytorch-nightly-cu128" }
 torch = { index = "pytorch-nightly-cu128" }
+torchstore = { git = "ssh://[email protected]/meta-pytorch/torchstore.git" }
 #vllm = { index = "vllm-nightly" }
 
 [tool.uv]