Cleanup deprecated usages of package monarch.utils in torchft

Summary:
Package `monarch.utils` was deprecated in 0.2 and removed in 0.4.
Migrate to the API it forwarded to.

Differential Revision: D93802250

Author: dulinriley

examples/monarch/train_distributed.py

@@ -16,7 +16,7 @@
 import torch
 from monarch.actor import Actor, current_rank, endpoint, HostMesh, ProcMesh, this_host
 from monarch.job import SlurmJob
-from monarch.utils import setup_env_for_distributed
+from monarch.spmd import setup_torch_elastic_env_async
 from torchtitan.config import ConfigManager, JobConfig
 from torchtitan.tools.logging import init_logger, logger
 from torchtitan.train import Trainer
@@ -157,7 +157,7 @@ async def start_replica(self) -> None:
 
         async with trainers_proc_mesh:
             await trainers_proc_mesh.logging_option(stream_to_client=True)
-            await setup_env_for_distributed(trainers_proc_mesh)
+            await setup_torch_elastic_env_async(trainers_proc_mesh)
 
             training_actors = trainers_proc_mesh.spawn(
                 "training_actors",