style: apply ruff auto-fixes (import order, unused/duplicate imports)

Applies ruff check --fix for I001, F401, and F811. The 8 remaining
F821 undefined-name errors in tests/test_externalize.py need a manual fix.

Author: DawerG

docs/coreai-core/tutorials/construct-a-graph.ipynb

@@ -76,11 +76,10 @@
     "\n",
     "import numpy as np\n",
     "\n",
-    "from coreai.authoring import AIModelAsset, AIProgram, Module, TensorSpec\n",
-    "\n",
     "# Graph-building primitives — pending re-export from coreai.authoring.\n",
     "from coreai._compiler.dialects import coreai as ops\n",
-    "from coreai._compiler.ir import Value"
+    "from coreai._compiler.ir import Value\n",
+    "from coreai.authoring import AIModelAsset, AIProgram, Module, TensorSpec"
    ]
   },
   {

docs/coreai-core/tutorials/run-an-aimodel.ipynb

@@ -64,7 +64,6 @@
     "from pathlib import Path\n",
     "\n",
     "import numpy as np\n",
-    "\n",
     "from coreai.authoring import AIModelAsset\n",
     "from coreai.runtime import InferenceFunction, NDArray\n",
     "\n",
@@ -108,10 +107,10 @@
     "from shutil import rmtree\n",
     "from typing import Annotated\n",
     "\n",
-    "from coreai.authoring import AIProgram, Module, TensorSpec\n",
     "# Pending re-export from coreai.authoring; see the previous tutorial.\n",
     "from coreai._compiler.dialects import coreai as ops\n",
     "from coreai._compiler.ir import Value\n",
+    "from coreai.authoring import AIProgram, Module, TensorSpec\n",
     "\n",
     "# Reconstruct asset.\n",
     "if asset_path.exists():\n",

docs/getting-started/quickstart.ipynb

@@ -68,7 +68,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import torch\n",
     "\n",
     "example_input = (torch.randn(1, 10),)\n",
     "exported = torch.export.export(model, args=example_input)"
@@ -145,10 +144,10 @@
    "outputs": [],
    "source": [
     "import tempfile\n",
-    "import torch\n",
     "from pathlib import Path\n",
     "\n",
     "import numpy as np\n",
+    "import torch\n",
     "from coreai.runtime import NDArray\n",
     "\n",
     "\n",
@@ -240,7 +239,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import tempfile\n",
     "\n",
     "\n",
     "async def run():\n",
@@ -274,6 +272,7 @@
    "outputs": [],
    "source": [
     "import torch\n",
+    "\n",
     "import coreai_torch\n",
     "\n",
     "model = SimpleModel().eval()\n",

docs/guides/composite-ops.ipynb

@@ -97,9 +97,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import torch\n",
-    "import torch.nn as nn\n",
-    "from coreai_torch.composite_ops import RMSNormImpl\n",
     "\n",
     "\n",
     "class RMSNorm(nn.Module):\n",
@@ -149,7 +146,6 @@
    "outputs": [],
    "source": [
     "import torch\n",
-    "import coreai_torch\n",
     "\n",
     "coreai_program = (\n",
     "    TorchConverter()\n",
@@ -185,10 +181,10 @@
    "outputs": [],
    "source": [
     "import tempfile\n",
-    "import torch\n",
     "from pathlib import Path\n",
     "\n",
     "import numpy as np\n",
+    "import torch\n",
     "from coreai.runtime import NDArray\n",
     "\n",
     "\n",

docs/guides/conversion-workflows.ipynb

@@ -86,6 +86,7 @@
    "outputs": [],
    "source": [
     "import torch\n",
+    "\n",
     "import coreai_torch\n",
     "\n",
     "model = MyModel().eval()\n",
@@ -133,7 +134,7 @@
    "source": [
     "import torch\n",
     "import torch.nn as nn\n",
-    "import coreai_torch\n",
+    "\n",
     "from coreai_torch import ExternalizeSpec, TorchConverter\n",
     "from coreai_torch.composite_ops import RMSNormImpl\n",
     "\n",

docs/guides/custom-metal-kernels.ipynb

@@ -122,6 +122,7 @@
    "outputs": [],
    "source": [
     "import torch\n",
+    "\n",
     "from coreai_torch import get_decomp_table\n",
     "\n",
     "model = AddModel().eval()\n",
@@ -182,6 +183,7 @@
    "source": [
     "import torch\n",
     "\n",
+    "\n",
     "def torch_matmul(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:\n",
     "    return torch.matmul(x, y)\n",
     "\n",
@@ -222,6 +224,7 @@
     "import torch\n",
     "import torch.nn as nn\n",
     "\n",
+    "\n",
     "def torch_sincos(x: torch.Tensor) -> list[torch.Tensor]:\n",
     "    return [torch.sin(x), torch.cos(x)]\n",
     "\n",

docs/guides/custom-op-lowering.ipynb

@@ -92,6 +92,7 @@
    "outputs": [],
    "source": [
     "import torch\n",
+    "\n",
     "from coreai_torch import get_decomp_table\n",
     "\n",
     "model = ScaledAddModel().eval()\n",
@@ -198,9 +199,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import numpy as np\n",
     "import torch\n",
     "import torch.nn as nn\n",
-    "import numpy as np\n",
     "\n",
     "from coreai_torch._utils import get_operand\n",
     "\n",

docs/guides/externalization.ipynb

@@ -86,8 +86,6 @@
     "import torch\n",
     "import torch.nn as nn\n",
     "\n",
-    "from coreai_torch import ExternalizeSpec\n",
-    "\n",
     "\n",
     "class RMSNormComposite(nn.Module):\n",
     "    def __init__(self, axes=-1, eps=1e-5, version=1):\n",

tests/compression/test_compression.py

@@ -14,7 +14,6 @@
 import torch
 from coreai.authoring import AIProgram
 from coreai.runtime import AIModel, NDArray
-
 from torch import nn
 from torch.export.exported_program import ExportedProgram