feat(examples): added torch.compile to examples of pytorch and local

Signed-off-by: Ishtiyaque-Alam <[email protected]>

Author: Ishtiyaque-Alam

examples/local/local-container-mnist.ipynb

@@ -52,7 +52,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "train-function",
    "metadata": {},
    "outputs": [],
@@ -102,7 +102,9 @@
     "\n",
     "    # Create the model and load it into the device.\n",
     "    device = torch.device(f\"{device}:{local_rank}\")\n",
-    "    model = nn.parallel.DistributedDataParallel(Net().to(device))\n",
+    "    model=Net().to(device)\n",
+    "    model=torch.compile(model)\n",
+    "    model = nn.parallel.DistributedDataParallel(model)\n",
     "    optimizer = optim.SGD(model.parameters(), lr=0.01, momentum=0.5)\n",
     "\n",
     "    # Download FashionMNIST dataset only on local_rank=0 process.\n",