Update filename and paper title to align with the article

PiperOrigin-RevId: 555913385

muNet/NAX.ipynb → muNet/TransformerExpansions.ipynb

@@ -22,9 +22,9 @@
         "id": "7BQOCQE5tZiU"
       },
       "source": [
-        "# Neural Architecture EXpansion (NAX): Incremental scaling of Transformer-based Neural Networks\n",
+        "# Composable Function-preserving Expansions for Transformer Architectures\n",
         "\n",
-        "This notebook contains implementations of the six function-preserving transformations of transformer-based models proposed in \"Neural Architecture EXpansion (NAX): Incremental scaling of Transformer-based Neural Networks\". We provide a basic implementation of a generic transformer architecture and show that each transformation is function-preserving, both for individual architectural components and the whole transformer model, as well as for individual transformations and combinations of transformations.\n"
+        "This notebook contains implementations of the six function-preserving transformations of transformer-based models proposed in \"Composable Function-preserving Expansions for Transformer Architectures\". We provide a basic implementation of a generic transformer architecture and show that each transformation is function-preserving, both for individual architectural components and the whole transformer model, as well as for individual transformations and combinations of transformations.\n"
       ]
     },
     {