torchmonarch-nightly 2025.9.3__cp311-cp311-manylinux2014_x86_64.whl → 2025.9.5__cp311-cp311-manylinux2014_x86_64.whl

monarch/_src/actor/actor_mesh.py

@@ -529,14 +529,38 @@ def as_endpoint(
 
 
 class Accumulator(Generic[P, R, A]):
+    """
+    Accumulate the result of a broadcast invocation of an endpoint
+    across a sliced mesh.
+
+    Usage:
+            >>> counter = Accumulator(Actor.increment, 0, lambda x, y: x + y)
+    """
+
     def __init__(
         self, endpoint: Endpoint[P, R], identity: A, combine: Callable[[A, R], A]
     ) -> None:
+        """
+        Args:
+            endpoint: Endpoint to accumulate the result of.
+            identity: Initial value of the accumulated value before the first combine invocation.
+            combine: Lambda invoked for combining the result of the endpoint with the accumulated value.
+        """
         self._endpoint: Endpoint[P, R] = endpoint
         self._identity: A = identity
         self._combine: Callable[[A, R], A] = combine
 
     def accumulate(self, *args: P.args, **kwargs: P.kwargs) -> "Future[A]":
+        """
+        Accumulate the result of the endpoint invocation.
+
+        Args:
+            args: Arguments to pass to the endpoint.
+            kwargs: Keyword arguments to pass to the endpoint.
+
+        Returns:
+            Future that resolves to the accumulated value.
+        """
         gen: Generator[Future[R], None, None] = self._endpoint.stream(*args, **kwargs)
 
         async def impl() -> A:
@@ -550,7 +574,7 @@ class Accumulator(Generic[P, R, A]):
 
 class ValueMesh(MeshTrait, Generic[R]):
     """
-    Container of return values, indexed by rank.
+    A mesh that holds the result of an endpoint invocation.
     """
 
     def __init__(self, shape: Shape, values: List[R]) -> None:
@@ -561,6 +585,18 @@ class ValueMesh(MeshTrait, Generic[R]):
         return ValueMesh(shape, self._values)
 
     def item(self, **kwargs) -> R:
+        """
+        Get the value at the given coordinates.
+
+        Args:
+            kwargs: Coordinates to get the value at.
+
+        Returns:
+            Value at the given coordinate.
+
+        Raises:
+            KeyError: If invalid coordinates are provided.
+        """
         coordinates = [kwargs.pop(label) for label in self._labels]
         if kwargs:
             raise KeyError(f"item has extra dimensions: {list(kwargs.keys())}")
@@ -568,6 +604,12 @@ class ValueMesh(MeshTrait, Generic[R]):
         return self._values[self._ndslice.nditem(coordinates)]
 
     def items(self) -> Iterable[Tuple[Point, R]]:
+        """
+        Generator that returns values for the provided coordinates.
+
+        Returns:
+            Values at all coordinates.
+        """
         extent = self._shape.extent
         for i, rank in enumerate(self._shape.ranks()):
             yield Point(i, extent), self._values[rank]
@@ -596,14 +638,27 @@ def send(
     selection: Selection = "all",
 ) -> None:
     """
-    Fire-and-forget broadcast invocation of the endpoint across all actors in the mesh.
+        Fire-and-forget broadcast invocation of the endpoint across a given selection of the mesh.
+
+        This sends the message to all actors but does not wait for any result. Use the port provided to
+        send the response back to the caller.
 
-    This sends the message to all actors but does not wait for any result.
+    Args:
+        endpoint: Endpoint to invoke.
+        args: Arguments to pass to the endpoint.
+        kwargs: Keyword arguments to pass to the endpoint.
+        port: Handle to send the response to.
+        selection: Selection query representing a subset of the mesh.
     """
     endpoint._send(args, kwargs, port, selection)
 
 
 class Port(Generic[R]):
+    """
+    Handle used to send reliable in-order messages through a channel to
+    a PortReceiver.
+    """
+
     def __init__(
         self,
         port_ref: PortRef | OncePortRef,
@@ -615,6 +670,13 @@ class Port(Generic[R]):
         self._rank = rank
 
     def send(self, obj: R) -> None:
+        """
+            Fire-and-forget send R-typed objects in order
+            through a channel to its corresponding PortReceiver.
+
+        Args:
+            obj: R-typed object to send.
+        """
         self._port_ref.send(
             self._mailbox,
             PythonMessage(PythonMessageKind.Result(self._rank), _pickle(obj)),
@@ -656,8 +718,17 @@ T = TypeVar("T")
 # not part of the Endpoint API because they way it accepts arguments
 # and handles concerns is different.
 class Channel(Generic[R]):
+    """
+    An advanced low level API for a communication channel used for message passing
+    between actors.
+
+    Provides static methods to create communication channels with port pairs
+    for sending and receiving messages of type R.
+    """
+
     @staticmethod
     def open(once: bool = False) -> Tuple["Port[R]", "PortReceiver[R]"]:
+        """ """
         mailbox = context().actor_instance._mailbox
         handle, receiver = mailbox.open_once_port() if once else mailbox.open_port()
         port_ref = handle.bind()
@@ -673,6 +744,14 @@ class Channel(Generic[R]):
 
 
 class PortReceiver(Generic[R]):
+    """
+    Receiver for messages sent through a communication channel.
+
+    Handles receiving R-typed objects sent from a corresponding Port.
+    Asynchronously message reception with optional supervision
+    monitoring for error handling.
+    """
+
     def __init__(
         self,
         mailbox: Mailbox,
@@ -784,6 +863,7 @@ class _Actor:
                     ins.rank = ctx.message_rank
                     try:
                         self.instance = Class(*args, **kwargs)
+                        self._maybe_exit_debugger()
                     except Exception as e:
                         self._saved_error = ActorError(
                             e, f"Remote actor {Class}.__init__ call failed."
@@ -956,6 +1036,15 @@ class Actor(MeshTrait, DeprecatedNotAFuture):
 
 
 class ActorMesh(MeshTrait, Generic[T], DeprecatedNotAFuture):
+    """
+    A group of actor instances of the same class.
+
+    Represents a collection of T-typed actor instances spawned at most once per process
+    that can be communicated with collectively or individually. Provides
+    methods for spawning actors, managing their lifecycle, and creating
+    endpoints for method invocation across the mesh.
+    """
+
     def __init__(
         self,
         Class: Type[T],

monarch/_src/actor/host_mesh.py

@@ -35,11 +35,23 @@ def this_proc() -> "ProcMesh":
 
 
 def create_local_host_mesh() -> "HostMesh":
+    """
+    Create a local host mesh for the current machine.
+
+    Returns:
+        HostMesh: A single-host mesh configured for local process allocation.
+    """
     cmd, args, env = _get_bootstrap_args()
     return HostMesh(Shape.unity(), ProcessAllocator(cmd, args, env))
 
 
 class HostMesh(MeshTrait):
+    """
+    HostMesh represents a collection of compute hosts that can be used to spawn
+    processes and actors. The class requires you to provide your AllocateMixin that
+    interfaces with the underlying resource allocator of your choice.
+    """
+
     def __init__(self, shape: Shape, allocator: AllocateMixin):
         self._allocator = allocator
         self._shape = shape
@@ -57,12 +69,14 @@ class HostMesh(MeshTrait):
         """
         Start new processes on this host mesh. By default this starts one proc
         on each host in the mesh. Additional procs can be started using `per_host` to
-        specify the local shape, e.g.
+        specify the local shape, e.g.`
             per_host = {'gpus': 8}
         Will create a proc mesh with an additional 'gpus' dimension.
 
         `bootstrap` is a function that will be run at startup on each proc and can be used to e.g.
         configure CUDA or NCCL. We guarantee that CUDA has not been initialized before boostrap is called.
+
+        TODO: For now, a new allocator is created for every new ProcMesh.
         """
         if per_host is None:
             per_host = {}
@@ -107,6 +121,12 @@ class HostMesh(MeshTrait):
 
 
 def fake_in_process_host() -> "HostMesh":
+    """
+    Create a host mesh for testing and development using a local allocator.
+
+    Returns:
+        HostMesh: A host mesh configured with local allocation for in-process use.
+    """
     return HostMesh(Shape.unity(), LocalAllocator())
 
 

monarch/_src/actor/proc_mesh.py

@@ -95,14 +95,10 @@ if TYPE_CHECKING:
 
 class SetupActor(Actor):
     """
-    A helper actor to setup the proc mesh with user defined setup method.
-    Typically used to setup the environment variables.
+    A helper actor to set up the actor mesh with user defined setup method.
     """
 
     def __init__(self, env: Callable[[], None]) -> None:
-        """
-        Initialize the setup actor with the user defined setup method.
-        """
         self._setup_method = env
 
     @endpoint
@@ -133,8 +129,12 @@ def _use_standin_mesh() -> bool:
     return os.getenv("USE_STANDIN_ACTOR_MESH", default="0") != "0"
 
 
-# Ultra-hack to allow actors to identify proc meshes but with no real functionality.
 class ProcMeshRef:
+    """
+    A serializable remote reference to a ProcMesh. The reference is weak: No support
+    for refcount'ing. Spawning actors on a ProcMeshRef a stopped or a failed mesh will fail.
+    """
+
     def __init__(self, proc_mesh_id: int) -> None:
         self._proc_mesh_id = proc_mesh_id
         self._host_mesh: Optional["HostMesh"] = None
@@ -179,6 +179,17 @@ def _deref_proc_mesh(proc_mesh: ProcMeshRef) -> "ProcMesh":
 
 
 class ProcMesh(MeshTrait, DeprecatedNotAFuture):
+    """
+    A distributed mesh of processes for actor computation.
+
+    ProcMesh represents a collection of processes that can spawn and manage actors.
+    It provides the foundation for distributed actor systems by managing process
+    allocation, lifecycle, and communication across multiple hosts and devices.
+
+    The ProcMesh supports spawning actors, monitoring process health, logging
+    configuration, and code synchronization across distributed processes.
+    """
+
     def __init__(
         self,
         hy_proc_mesh: "Shared[HyProcMesh]",
@@ -249,6 +260,22 @@ class ProcMesh(MeshTrait, DeprecatedNotAFuture):
         return pm
 
     def spawn(self, name: str, Class: Type[T], *args: Any, **kwargs: Any) -> T:
+        """
+        Spawn a T-typed actor mesh on the process mesh.
+
+        Args:
+        - `name`: The name of the actor.
+        - `Class`: The class of the actor to spawn.
+        - `args`: Positional arguments to pass to the actor's constructor.
+        - `kwargs`: Keyword arguments to pass to the actor's constructor.
+
+        Returns:
+        - The actor instance.
+
+        Usage:
+            >>> procs: ProcMesh = host_mesh.spawn_procs(per_host={"gpus": 8})
+            >>> counters: Counter = procs.spawn("counters", Counter, 0)
+        """
         if self._slice:
             raise NotImplementedError("NYI: spawn on slice of a proc mesh.")
         return self._spawn_nonblocking(name, Class, *args, **kwargs)
@@ -294,19 +321,9 @@ class ProcMesh(MeshTrait, DeprecatedNotAFuture):
         Allocate a process mesh according to the provided alloc.
         Returns when the mesh is fully allocated.
 
-        Arguments:
-        - `alloc`: The alloc to allocate according to.
+        Args:
+        - `alloc`: A generator that yields a list of allocations.
         - `setup`: An optional lambda function to configure environment variables on the allocated mesh.
-        Use the `current_rank()` method within the lambda to obtain the rank.
-
-        Example of a setup method to initialize torch distributed environment variables:
-        ```
-        def setup():
-            rank = current_rank()
-            os.environ["RANK"] = str(rank)
-            os.environ["WORLD_SIZE"] = str(len(rank.shape))
-            os.environ["LOCAL_RANK"] = str(rank["gpus"])
-        ```
         """
 
         async def task() -> HyProcMesh:
@@ -432,6 +449,14 @@ class ProcMesh(MeshTrait, DeprecatedNotAFuture):
         conda: bool = False,
         auto_reload: bool = False,
     ) -> None:
+        """
+        Sync local code changes to the remote processes.
+
+        Args:
+            workspace: The workspace to sync.
+            conda: If True, also sync the currently activated conda env.
+            auto_reload: If True, automatically reload the workspace on changes.
+        """
         if self._code_sync_client is None:
             self._code_sync_client = CodeSyncMeshClient.spawn_blocking(
                 proc_mesh=await self._proc_mesh_for_asyncio_fixme,
@@ -525,6 +550,10 @@ class ProcMesh(MeshTrait, DeprecatedNotAFuture):
         return self
 
     def stop(self) -> Future[None]:
+        """
+        This will stop all processes (and actors) in the mesh and
+        release any resources associated with the mesh.
+        """
         self._logging_manager.stop()
 
         async def _stop_nonblocking() -> None:
@@ -574,6 +603,23 @@ class ProcMesh(MeshTrait, DeprecatedNotAFuture):
 
 
 def local_proc_mesh(*, gpus: Optional[int] = None, hosts: int = 1) -> ProcMesh:
+    """
+    Create a local process mesh for testing and development.
+
+    This function creates a process mesh using local allocation instead of
+    distributed process allocation. Primarily used for testing scenarios.
+
+    Args:
+        gpus: Number of GPUs to allocate per host. If None, uses local device count.
+        hosts: Number of hosts to allocate. Defaults to 1.
+
+    Returns:
+        ProcMesh: A locally allocated process mesh.
+
+    Warning:
+        This function is deprecated. Use `fake_in_process_host().spawn_procs()`
+        for testing or `this_proc().spawn_procs()` for current process actors.
+    """
     warnings.warn(
         "Use monarch._src.actor.host_mesh.fake_in_process_host().spawn_procs for testing. For launching an actor in the current process use this_proc().spawn_procs()",
         DeprecationWarning,
@@ -596,6 +642,22 @@ def sim_proc_mesh(
     dcs: int = 1,
     regions: int = 1,
 ) -> ProcMesh:
+    """Create a simulated process mesh for testing distributed scenarios.
+
+    This function creates a process mesh using simulation allocation to test
+    distributed behavior without requiring actual remote resources.
+
+    Args:
+        gpus: Number of GPUs per host. Defaults to 1.
+        hosts: Number of hosts. Defaults to 1.
+        racks: Number of racks. Defaults to 1.
+        zones: Number of zones. Defaults to 1.
+        dcs: Number of data centers. Defaults to 1.
+        regions: Number of regions. Defaults to 1.
+
+    Returns:
+        ProcMesh: A simulated process mesh with the specified topology.
+    """
     spec: AllocSpec = AllocSpec(
         AllocConstraints(),
         hosts=hosts,
@@ -658,6 +720,25 @@ def proc_mesh(
     env: dict[str, str] | None = None,
     setup: Callable[[], None] | None = None,
 ) -> ProcMesh:
+    """
+    Create a distributed process mesh across hosts.
+
+    This function creates a process mesh using distributed process allocation
+    across multiple hosts and GPUs. Used for production distributed computing.
+
+    Args:
+        gpus: Number of GPUs per host. If None, uses local device count.
+        hosts: Number of hosts to allocate. Defaults to 1.
+        env: Environment variables to set on remote processes.
+        setup: Optional setup function to run on each process at startup.
+
+    Returns:
+        ProcMesh: A distributed process mesh with the specified configuration.
+
+    Warning:
+        This function is deprecated. Use `this_host().spawn_procs()` with
+        appropriate per_host configuration instead.
+    """
     warnings.warn(
         "use this_host().spawn_procs(per_host = {'hosts': 2, 'gpus': 3}) instead of monarch.actor.proc_mesh(hosts=2, gpus=3)",
         DeprecationWarning,

monarch/actor/__init__.py

@@ -9,6 +9,7 @@
 Monarch Actor API - Public interface for actor functionality.
 """
 
+from monarch._rust_bindings.monarch_hyperactor.shape import Extent
 from monarch._src.actor.actor_mesh import (
     Accumulator,
     Actor,
@@ -19,6 +20,7 @@ from monarch._src.actor.actor_mesh import (
     current_actor_name,
     current_rank,
     current_size,
+    Endpoint,
     Point,
     Port,
     PortReceiver,
@@ -70,4 +72,6 @@ __all__ = [
     "hosts_from_config",
     "Port",
     "PortReceiver",
+    "Endpoint",
+    "Extent",
 ]

monarch/common/tensor.py

@@ -80,10 +80,6 @@ class Tensor(Referenceable, BaseTensor):
     in a device mesh. It provides the same interface as PyTorch tensors but
     enables distributed operations and communication patterns.
 
-    Args:
-        fake (torch.Tensor): A fake tensor representing the shape and type
-        mesh (DeviceMesh): The device mesh this tensor is distributed across
-        stream (Stream): The computation stream for this tensor
     """
 
     # pyre-fixme[13]: Attribute `stream` is never initialized.

monarch/tools/components/hyperactor.py

@@ -9,7 +9,8 @@ import getpass
 from typing import Optional
 
 from monarch.tools import mesh_spec
-from monarch.tools.config import UnnamedAppDef
+
+from monarch.tools.config import NOT_SET
 from monarch.tools.mesh_spec import mesh_spec_from_str
 from torchx import specs
 
@@ -19,6 +20,7 @@ _USER: str = getpass.getuser()
 
 DEFAULT_NAME: str = f"monarch-{_USER}"
 
+
 __version__ = "latest"  # TODO get version from monarch.__version_
 
 
@@ -28,7 +30,7 @@ def host_mesh(
     env: Optional[dict[str, str]] = None,
     port: int = mesh_spec.DEFAULT_REMOTE_ALLOCATOR_PORT,
     program: str = "monarch_bootstrap",  # installed with monarch wheel (as console script)
-) -> UnnamedAppDef:
+) -> specs.AppDef:
     """
     Args:
         name: the name of the monarch server job
@@ -39,7 +41,7 @@ def host_mesh(
         program: path to the binary that the remote process allocator spawns on an allocation request
     """
 
-    appdef = UnnamedAppDef()
+    appdef = specs.AppDef(name=NOT_SET)
 
     for mesh in [mesh_spec_from_str(mesh) for mesh in meshes]:
         mesh_role = specs.Role(

monarch/tools/config/__init__.py

@@ -7,26 +7,22 @@
 # pyre-strict
 import warnings
 from dataclasses import dataclass, field
-from typing import Any, Dict, List, TYPE_CHECKING
+from typing import Any
 
 from monarch.tools.config.workspace import Workspace
 
-# Defer the import of Role to avoid requiring torchx at import time
-if TYPE_CHECKING:
-    from torchx.specs import Role
-
+# Gracefully handle cases where torchx might not be installed
+# NOTE: this can be removed once torchx.specs moves to monarch.session
+try:
+    from torchx import specs
+except ImportError:
+    pass
 
 NOT_SET: str = "__NOT_SET__"
 
 
-@dataclass
-class UnnamedAppDef:
-    """
-    A TorchX AppDef without a name.
-    """
-
-    roles: List["Role"] = field(default_factory=list)
-    metadata: Dict[str, str] = field(default_factory=dict)
+def _empty_appdef() -> "specs.AppDef":
+    return specs.AppDef(name=NOT_SET)
 
 
 @dataclass
@@ -39,7 +35,7 @@ class Config:
     scheduler_args: dict[str, Any] = field(default_factory=dict)
     workspace: Workspace = field(default_factory=Workspace.null)
     dryrun: bool = False
-    appdef: UnnamedAppDef = field(default_factory=UnnamedAppDef)
+    appdef: "specs.AppDef" = field(default_factory=_empty_appdef)
 
     def __post_init__(self) -> None:
         # workspace used to be Optional[str]

monarch/tools/config/defaults.py

@@ -12,7 +12,7 @@ import warnings
 from typing import Callable
 
 from monarch.tools.components import hyperactor
-from monarch.tools.config import Config, UnnamedAppDef
+from monarch.tools.config import Config
 from monarch.tools.config.workspace import Workspace
 
 from torchx import specs
@@ -25,7 +25,7 @@ from torchx.schedulers import (
 )
 
 
-def component_fn(scheduler: str) -> Callable[..., UnnamedAppDef]:
+def component_fn(scheduler: str) -> Callable[..., specs.AppDef]:
     """The default TorchX component function for the scheduler"""
     return hyperactor.host_mesh
 

monarch/tools/mesh_spec.py

@@ -9,8 +9,6 @@ import string
 from dataclasses import dataclass, field
 from typing import Any, Optional
 
-from monarch.tools.config import UnnamedAppDef
-
 from monarch.tools.network import get_sockaddr
 from torchx import specs
 from torchx.specs.api import is_terminal
@@ -72,7 +70,7 @@ def _tag(mesh_name: str, tag_template: str) -> str:
     return string.Template(tag_template).substitute(mesh_name=mesh_name)
 
 
-def tag_as_metadata(mesh_spec: MeshSpec, appdef: UnnamedAppDef) -> None:
+def tag_as_metadata(mesh_spec: MeshSpec, appdef: specs.AppDef) -> None:
     appdef.metadata[_tag(mesh_spec.name, _TAG_HOST_TYPE)] = mesh_spec.host_type
     appdef.metadata[_tag(mesh_spec.name, _TAG_GPUS)] = str(mesh_spec.gpus)
     appdef.metadata[_tag(mesh_spec.name, _TAG_TRANSPORT)] = mesh_spec.transport

{torchmonarch_nightly-2025.9.3.dist-info → torchmonarch_nightly-2025.9.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: torchmonarch-nightly
-Version: 2025.9.3
+Version: 2025.9.5
 Summary: Monarch: Single controller library
 Author: Meta
 Author-email: oncall+monarch@xmail.facebook.com
@@ -9,6 +9,7 @@ Requires-Python: >= 3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: torch
+Requires-Dist: torchshow
 Requires-Dist: pyzmq
 Requires-Dist: requests
 Requires-Dist: numpy
@@ -64,7 +65,7 @@ fut.get()
 ```
 
 
-The [introduction to monarch concepts](getting_started.html) provides an introduction to using these features.
+The [introduction to monarch concepts](https://meta-pytorch.org/monarch/generated/examples/getting_started.html) provides an introduction to using these features.
 
 > ⚠️ **Early Development Warning** Monarch is currently in an experimental
 > stage. You should expect bugs, incomplete features, and APIs that may change
@@ -73,18 +74,24 @@ The [introduction to monarch concepts](getting_started.html) provides an introdu
 > work. It's recommended that you signal your intention to contribute in the
 > issue tracker, either by filing a new issue or by claiming an existing one.
 
-Note: Monarch is currently only supported on Linux systems
-
 ## 📖 Documentation
 
 View Monarch's hosted documentation [at this link](https://meta-pytorch.org/monarch/).
 
 ## Installation
+Note for running distributed tensors, the local torch version must match the version that monarch was built with.
 
 ### On Fedora distributions
+## Stable
+`pip install torchmonarch`
+
+torchmonarch stable is built with the latest stable torch.
 
+## Nightly
 `pip install torchmonarch-nightly`
 
+torchmonarch-nightly is built with torch nightly.
+
 or manually
 
 ```sh

{torchmonarch_nightly-2025.9.3.dist-info → torchmonarch_nightly-2025.9.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 monarch/__init__.py,sha256=mgKiyD1kxky-1pvhMlNfF4VmxWnhi-FSYZNFzkW1BEM,7052
-monarch/_rust_bindings.so,sha256=V26PTUdu3YGZbYGLq-vcGxhxMhNpG2A4waq8dUhiKRs,61340152
+monarch/_rust_bindings.so,sha256=4PSOldY67JPiHo6-GBN1Cd5zhbVOxcCU5sTSrMACKAc,61372560
 monarch/_testing.py,sha256=5BDMVA4hBMo780rsJ39vRmUZi6mTN8aYY7I9grJRjJ8,7841
 monarch/actor_mesh.py,sha256=VtPU9syi_vUdwDSJJ639Z4Y_EcWZUScyoj0lQ88RQPs,421
 monarch/bootstrap_main.py,sha256=39OZpNMrfvvNJf-iwuNzgslzYA_ItaRPHfXGn_V74N0,524
@@ -8,7 +8,7 @@ monarch/fetch.py,sha256=CssP25dMqyJnJAWoC41lwkMnSbvS-f2DL9PRbudJXfc,1704
 monarch/gradient_generator.py,sha256=b7PmoN_F3c5hQglfHeW_v5htYnePKvJGkzZN-tpHR4A,6396
 monarch/memory.py,sha256=ol86dBhFAJqg78iF25-BuK0wuwj1onR8FIioZ_B0gjw,1377
 monarch/mesh_controller.py,sha256=Y_26Cnmp72TccNbWdDQhq18j7de7pSw83E_fREJX9Yo,15372
-monarch/monarch_controller,sha256=LVqFQjApcnCW9AWWvgSrIdggGFtJJ9hh0Cf2yU1Qxhs,32390232
+monarch/monarch_controller,sha256=8pl9HaAiJvsa1X8MsUQnbVKiD6oClL730oHsUUdkzwo,32446984
 monarch/notebook.py,sha256=zu9MKDFKf1-rCM2TqFSRJjMBeiWuKcJSyUFLvoZRQzs,25949
 monarch/opaque_module.py,sha256=jCcg0DjbcEVXA9WNG0NhUzGteLHOJLTZEBvrIYJIAns,10436
 monarch/opaque_object.py,sha256=x1LoX6RIMGh4ux52xIfhPgoh6PhZHdkf9bMccHW3DW0,2808
@@ -25,17 +25,17 @@ monarch/tensorboard.py,sha256=MnLgH5lbqeUJauEuirEgR6L_qYl2NGdtwZOWIAuOZao,2587
 monarch/world_mesh.py,sha256=ob5dJWaC49Uw0xqClHBm8CQLvL4xKnjd4TGzk7k8NxI,980
 monarch/_src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monarch/_src/actor/__init__.py,sha256=4iK3nzQZmEPe0HGNY70fABBenw3lCVVaaF0xddF5Fa0,235
-monarch/_src/actor/actor_mesh.py,sha256=Y4IaWtVwL4h8nZ2qpOEIs3sqI2uCpVwYQZUCCHUSO80,37707
+monarch/_src/actor/actor_mesh.py,sha256=P9b4CvxYeYfJecPxeEtohAasvGgzzElcvxh9chALeAA,40526
 monarch/_src/actor/allocator.py,sha256=UVGhrkPQMqPQp6vUngPI361s6yCEfZ0gfz8WTtG2om4,9392
 monarch/_src/actor/bootstrap_main.py,sha256=7T7ARumcHLZ5RI-k5jej9tBil0J7-BUSVFKwAZO2tJU,2413
 monarch/_src/actor/device_utils.py,sha256=gBpl23wMjppVAEzzj8U9HyX-B7Bs2_3ftiMAkzUS4j4,577
 monarch/_src/actor/endpoint.py,sha256=_VaPHc0Fcj1P5nDzUXt8xnS6iw7-HO6hGx7W5RtU3eU,10916
 monarch/_src/actor/event_loop.py,sha256=2i4fKIkemBzua_t47BqVa2roZ6fWB6sbmMFPNx2zKN0,2832
 monarch/_src/actor/future.py,sha256=idgqzU_B5qWfClIP5dTLapmBflWq5va-ujAzUbT1Asc,7490
-monarch/_src/actor/host_mesh.py,sha256=xSto21T4ESnF55UfcQOkWZxsIiiv3VnOzyrcrZ_or7Y,4405
+monarch/_src/actor/host_mesh.py,sha256=8SOkg_LhHuzLyhpwxT7Yw1_h8QrIlwfWhrSwHyAvfnk,5083
 monarch/_src/actor/logging.py,sha256=9aguohqCtvLVwWGTFM7o-rBptT26BjI2s6E5Of2imM4,3311
 monarch/_src/actor/pickle.py,sha256=FhdbAEsGrsD7f25bxF7HlROLm6j2TTvmToq8P1kyhB8,2913
-monarch/_src/actor/proc_mesh.py,sha256=fijaxhvVn1adtPFPaZ_-16pvWo1-rhjbB6qHkW2dAkk,24779
+monarch/_src/actor/proc_mesh.py,sha256=lYrRMQNOGAdFXuFvc3lQ68xIS01YJWMkpi8qH5HHAHE,27791
 monarch/_src/actor/python_extension_methods.py,sha256=QujLWOQQbDdGCin8tZfDxyIwkM-Md4t9QtcTGTHOE_s,3493
 monarch/_src/actor/shape.py,sha256=PJqxpQEISHlxK8rrlKWpcNMEHiGxBbc6TsHcGZCOsyE,8472
 monarch/_src/actor/source_loader.py,sha256=TGHmExLyxPDcCyuG254zo6aUqHMpl-j0VWzxa9rkJYQ,1405
@@ -52,7 +52,7 @@ monarch/_src/debug_cli/__init__.py,sha256=NNrKh5KdiYdbxOhin8x-gw_-tvcuGex2UbS_z7
 monarch/_src/debug_cli/debug_cli.py,sha256=OJqqVFXcMkj-bnrxcE2VnjIgA5xrlKjEtCstrsdPcm0,1146
 monarch/_src/tensor_engine/__init__.py,sha256=Md3cCHD7Ano9kV15PqGbicgUO-RMdh4aVy1yKiDt_xE,208
 monarch/_src/tensor_engine/rdma.py,sha256=62saqcXo6oUxH9rEZShNGLYdRZ_zizLQKhub7LDSaBg,8418
-monarch/actor/__init__.py,sha256=9CXerFsBVR-lcpSEw_lIsNNP0qqCWoSA9exyDhhGvHI,1458
+monarch/actor/__init__.py,sha256=hHf8ri2czQwi-Z23Z9cYZ2FvkVbYOcDA_GTLW_rju7k,1569
 monarch/builtins/__init__.py,sha256=QcfnHZGbc2qktBg7DyZt2ruE6VahnIt4S8lEZLHdJqU,443
 monarch/builtins/log.py,sha256=H1QkuVzwxyi36Zyv-XR0VN0QsNimBWwxE1__fjs0_2o,554
 monarch/builtins/random.py,sha256=wPbvscg7u53EXpMFo885fO2XOlsyjrNAJ4rBxLzfxdg,1839
@@ -84,7 +84,7 @@ monarch/common/reference.py,sha256=O26lkzEeVwj0S1xEy-OLqdHVnACmmlbQCUmXRrW4n1Q,9
 monarch/common/remote.py,sha256=uc2JUbYHMnEZRnw9ZuS4mgvK_UHCuRaIIFbzZpx89hQ,11485
 monarch/common/selection.py,sha256=lpWFbZs3ArYy29e-53eoAVAjQFksf1RvZz9NvM0CUW4,308
 monarch/common/stream.py,sha256=_ejoxafHtdD10lLzznRCXKwrkZ_ZH9k_VTgiA5yfBrI,3583
-monarch/common/tensor.py,sha256=tqoXyvxrmfTtOm6ToWoy9NVg1NCjlt1KPjrNtYOMCco,29824
+monarch/common/tensor.py,sha256=9FIUn5--VeacYWhEhMvysdG0yc_zq4eW3X3526RCw3w,29598
 monarch/common/tensor_factory.py,sha256=qm8NZx-5ezMAFjNLiXQvb66okm5XgdboB_GRarGOdN0,801
 monarch/common/tree.py,sha256=1DG3siiE7ixBV6v5cwN8RT_17aJhYZTE-L3i7wZe2_c,2282
 monarch/controller/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
@@ -98,7 +98,7 @@ monarch/debug_cli/__init__.py,sha256=NNrKh5KdiYdbxOhin8x-gw_-tvcuGex2UbS_z7MV9g0
 monarch/debug_cli/__main__.py,sha256=FGsQn54RkC_3gpRrm_UFrGiDDHRbMeGzXXsGANr5UHU,317
 monarch/gradient/__init__.py,sha256=kqmzwt16mMpk0M3GhpgP_f7da4DGnaV9chDzbt66k4Q,308
 monarch/gradient/_gradient_generator.pyi,sha256=6cX0UxaDt9NAlwgIhTgnweqGOf6qRhHiGnUzSWNCxdU,630
-monarch/gradient/_gradient_generator.so,sha256=XfxpsBGsvv3LclZWw5tYpV_y9ZUL2erSKOSVD8pzK9M,12172008
+monarch/gradient/_gradient_generator.so,sha256=kGzXSoii5ODi8ZA7gz9D38Lt5sxGi5POXcGYmidS0-Q,12174336
 monarch/parallel/__init__.py,sha256=6920kIkhiX7AiyjYvyc1ad8ccP-bStJJ1sS5KkeN2P0,352
 monarch/parallel/pipelining/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
 monarch/parallel/pipelining/runtime.py,sha256=KK8TG1gUYEzSsquiZoPTWGSIC74mlncD7cYknKxfb3c,32470
@@ -127,13 +127,13 @@ monarch/tools/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
 monarch/tools/cli.py,sha256=b3mKZnK-MwP7JwskTxHI0KcJXxSU6498jEb2ntVr_VM,5001
 monarch/tools/colors.py,sha256=XrBkslKoaoDeXqiTluiiuvFLYd-weKp1sjw7DYWz2RY,581
 monarch/tools/commands.py,sha256=z4vCPtn_Ypic7L4_Jd3nMJWyyE4olUPqDe4cpJsDKZ4,13873
-monarch/tools/mesh_spec.py,sha256=9CMiLWNrgSUF0oWoAtmUT2CweMgqi4pLI18w7-EZWxs,8032
+monarch/tools/mesh_spec.py,sha256=lkKZ7RxuJKY19X6kdiU_V6IWlH1GHidynOaTbuCOsAY,7983
 monarch/tools/network.py,sha256=mN8Fx9mervxM3VdFHRn4ZXt4z7yWxZp52BTxx2tfpus,2455
 monarch/tools/utils.py,sha256=gcZyalfoBC6Y3v65h-QMngwXsn24ejXh2TH8RxlgXkA,1888
 monarch/tools/components/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
-monarch/tools/components/hyperactor.py,sha256=P_5XTbCl-niCj7wua-9iDaEuLVhYohetT3nScWgA7Uc,2169
-monarch/tools/config/__init__.py,sha256=3VV8zWKUc-yKA-wybCIbwSvDjtIzCS4dHX9KChvxn9A,2296
-monarch/tools/config/defaults.py,sha256=BRWDbfUkPX1Mz982EzRsYn-DlQpQw2kLIwVV2lbhpnE,2203
+monarch/tools/components/hyperactor.py,sha256=OR5JtH2UCao3ke3vMohzVbuo_L0gZ_jTw8ud82qLj3M,2175
+monarch/tools/config/__init__.py,sha256=wCw2qwGJL1gFuo9Wpvnrva6NKDLyjf2Yglm6Q9UJYkI,2224
+monarch/tools/config/defaults.py,sha256=twUF6eT9HjJyxEZYrz2SoROHHXi3YPUDSeAelJRLBSU,2187
 monarch/tools/config/environment.py,sha256=ikEZKATa2e_8h9pN4_3TzhIHWb4ZZfRT5XtOVoOmHjI,1628
 monarch/tools/config/workspace.py,sha256=a2YzFBTLUB_VrO3kt6dCV5TlmhCH4LyRX3JCMzu7Iv0,6049
 monarch/utils/__init__.py,sha256=9ofjBGAMZo1VGsn7ufiDlrVheMw4Ye34p-isDfveUxc,295
@@ -186,9 +186,9 @@ tests/simulator/test_profiling.py,sha256=TGYCfzTLdkpIwnOuO6KApprmrgPIRQe60KRX3wk
 tests/simulator/test_simulator.py,sha256=LO8lA0ssY-OGEBL5ipEu74f97Y765TEwfUOv-DtIptM,14568
 tests/simulator/test_task.py,sha256=ipqBDuDAysuo1xOB9S5psaFvwe6VATD43IovCTSs0t4,2327
 tests/simulator/test_worker.py,sha256=QrWWIJ3HDgDLkBPRc2mwYPlOQoXQcj1qRfc0WUfKkFY,3507
-torchmonarch_nightly-2025.9.3.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
-torchmonarch_nightly-2025.9.3.dist-info/METADATA,sha256=L5a1VhSSe2RevR33LVzvECh6LjPdlCV_bWkaT1nD7Zk,6184
-torchmonarch_nightly-2025.9.3.dist-info/WHEEL,sha256=JC9FVdjbTDi9l3EyrqUd11CgmN9LkBi1g5dFHayafwA,104
-torchmonarch_nightly-2025.9.3.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
-torchmonarch_nightly-2025.9.3.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
-torchmonarch_nightly-2025.9.3.dist-info/RECORD,,
+torchmonarch_nightly-2025.9.5.dist-info/licenses/LICENSE,sha256=e0Eotbf_rHOYPuEUlppIbvwy4SN98CZnl_hqwvbDA4Q,1530
+torchmonarch_nightly-2025.9.5.dist-info/METADATA,sha256=6WKzpl0pCJnxLhSxTckSsA7C5ncuUZJ2_NzujIuxhiQ,6474
+torchmonarch_nightly-2025.9.5.dist-info/WHEEL,sha256=JC9FVdjbTDi9l3EyrqUd11CgmN9LkBi1g5dFHayafwA,104
+torchmonarch_nightly-2025.9.5.dist-info/entry_points.txt,sha256=60QVSpYVzkzS4iDOiLp0fsLxVp47X3J2l3v7W-59LMo,117
+torchmonarch_nightly-2025.9.5.dist-info/top_level.txt,sha256=E-ZssZzyM17glpVrh-S9--qJ-w9p2EjuYOuNw9tQ4Eg,33
+torchmonarch_nightly-2025.9.5.dist-info/RECORD,,