rebase

2025-09-18 18:28:12 +08:00 · 2025-01-09 21:56:20 +00:00
parent 04b749a588
commit d8d647015b
2642 changed files with 137687 additions and 70861 deletions
--- a/docs/build/html/genindex.html
+++ b/docs/build/html/genindex.html
@@ -7,7 +7,7 @@
  <head>
    <meta charset="utf-8" />
    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Index &#8212; MLX 0.21.1 documentation</title>
+    <title>Index &#8212; MLX 0.22.0 documentation</title>
  
  
  
@@ -15,8 +15,8 @@
    document.documentElement.dataset.mode = localStorage.getItem("mode") || "";
    document.documentElement.dataset.theme = localStorage.getItem("theme") || "";
  </script>
-  <!-- 
-    this give us a css class that will be invisible only if js is disabled 
+  <!--
+    this give us a css class that will be invisible only if js is disabled
  -->
  <noscript>
    <style>
@@ -26,19 +26,19 @@
  </noscript>
  
  <!-- Loaded before other Sphinx assets -->
-  <link href="_static/styles/theme.css?digest=26a4bc78f4c0ddb94549" rel="stylesheet" />
-<link href="_static/styles/pydata-sphinx-theme.css?digest=26a4bc78f4c0ddb94549" rel="stylesheet" />
+  <link href="_static/styles/theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />
+<link href="_static/styles/pydata-sphinx-theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" />

-    <link rel="stylesheet" type="text/css" href="_static/pygments.css?v=fa44fd50" />
+    <link rel="stylesheet" type="text/css" href="_static/pygments.css?v=03e43079" />
    <link rel="stylesheet" type="text/css" href="_static/styles/sphinx-book-theme.css?v=a3416100" />
  
  <!-- So that users can add custom icons -->
-  <script src="_static/scripts/fontawesome.js?digest=26a4bc78f4c0ddb94549"></script>
+  <script src="_static/scripts/fontawesome.js?digest=8878045cc6db502f8baf"></script>
  <!-- Pre-loaded scripts that we'll load fully later -->
-  <link rel="preload" as="script" href="_static/scripts/bootstrap.js?digest=26a4bc78f4c0ddb94549" />
-<link rel="preload" as="script" href="_static/scripts/pydata-sphinx-theme.js?digest=26a4bc78f4c0ddb94549" />
+  <link rel="preload" as="script" href="_static/scripts/bootstrap.js?digest=8878045cc6db502f8baf" />
+<link rel="preload" as="script" href="_static/scripts/pydata-sphinx-theme.js?digest=8878045cc6db502f8baf" />

-    <script src="_static/documentation_options.js?v=acb17c73"></script>
+    <script src="_static/documentation_options.js?v=c952a61b"></script>
    <script src="_static/doctools.js?v=9a2dae69"></script>
    <script src="_static/sphinx_highlight.js?v=dc90522c"></script>
    <script src="_static/scripts/sphinx-book-theme.js?v=887ef09a"></script>
@@ -48,7 +48,7 @@
    <link rel="search" title="Search" href="search.html" />
  <meta name="viewport" content="width=device-width, initial-scale=1"/>
  <meta name="docsearch:language" content="en"/>
-  <meta name="docsearch:version" content="0.21.1" />
+  <meta name="docsearch:version" content="0.22.0" />
  </head>
  
  
@@ -129,8 +129,8 @@
      
    
    
-    <img src="_static/mlx_logo.png" class="logo__image only-light" alt="MLX 0.21.1 documentation - Home"/>
-    <img src="_static/mlx_logo_dark.png" class="logo__image only-dark pst-js-only" alt="MLX 0.21.1 documentation - Home"/>
+    <img src="_static/mlx_logo.png" class="logo__image only-light" alt="MLX 0.22.0 documentation - Home"/>
+    <img src="_static/mlx_logo_dark.png" class="logo__image only-dark pst-js-only" alt="MLX 0.22.0 documentation - Home"/>
  
  
 </a></div>
@@ -159,6 +159,7 @@
 <li class="toctree-l1"><a class="reference internal" href="usage/numpy.html">Conversion to NumPy and Other Frameworks</a></li>
 <li class="toctree-l1"><a class="reference internal" href="usage/distributed.html">Distributed Communication</a></li>
 <li class="toctree-l1"><a class="reference internal" href="usage/using_streams.html">Using Streams</a></li>
+<li class="toctree-l1"><a class="reference internal" href="usage/export.html">Exporting Functions</a></li>
 </ul>
 <p aria-level="2" class="caption" role="heading"><span class="caption-text">Examples</span></p>
 <ul class="nav bd-sidenav">
@@ -227,6 +228,7 @@
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.Dtype.html">mlx.core.Dtype</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.DtypeCategory.html">mlx.core.DtypeCategory</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.issubdtype.html">mlx.core.issubdtype</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.finfo.html">mlx.core.finfo</a></li>
 </ul>
 </details></li>
 <li class="toctree-l1 has-children"><a class="reference internal" href="python/devices_and_streams.html">Devices and Streams</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
@@ -241,6 +243,13 @@
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.synchronize.html">mlx.core.synchronize</a></li>
 </ul>
 </details></li>
+<li class="toctree-l1 has-children"><a class="reference internal" href="python/export.html">Export Functions</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.export_function.html">mlx.core.export_function</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.import_function.html">mlx.core.import_function</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.exporter.html">mlx.core.exporter</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.export_to_dot.html">mlx.core.export_to_dot</a></li>
+</ul>
+</details></li>
 <li class="toctree-l1 has-children"><a class="reference internal" href="python/ops.html">Operations</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.abs.html">mlx.core.abs</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.add.html">mlx.core.add</a></li>
@@ -323,6 +332,7 @@
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.isneginf.html">mlx.core.isneginf</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.isposinf.html">mlx.core.isposinf</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.issubdtype.html">mlx.core.issubdtype</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.kron.html">mlx.core.kron</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.left_shift.html">mlx.core.left_shift</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.less.html">mlx.core.less</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.less_equal.html">mlx.core.less_equal</a></li>
@@ -378,6 +388,8 @@
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.sign.html">mlx.core.sign</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.sin.html">mlx.core.sin</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.sinh.html">mlx.core.sinh</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.slice.html">mlx.core.slice</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.slice_update.html">mlx.core.slice_update</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.softmax.html">mlx.core.softmax</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.sort.html">mlx.core.sort</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.split.html">mlx.core.split</a></li>
@@ -402,6 +414,7 @@
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.tri.html">mlx.core.tri</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.tril.html">mlx.core.tril</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.triu.html">mlx.core.triu</a></li>
+<li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.unflatten.html">mlx.core.unflatten</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.var.html">mlx.core.var</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.view.html">mlx.core.view</a></li>
 <li class="toctree-l2"><a class="reference internal" href="python/_autosummary/mlx.core.where.html">mlx.core.where</a></li>
@@ -694,6 +707,7 @@
 <li class="toctree-l1"><a class="reference internal" href="dev/extensions.html">Custom Extensions in MLX</a></li>
 <li class="toctree-l1"><a class="reference internal" href="dev/metal_debugger.html">Metal Debugger</a></li>
 <li class="toctree-l1"><a class="reference internal" href="dev/custom_metal_kernels.html">Custom Metal Kernels</a></li>
+<li class="toctree-l1"><a class="reference internal" href="dev/mlx_in_cpp.html">Using MLX in C++</a></li>
 </ul>

    </div>
@@ -702,9 +716,14 @@
  
  
  <div class="sidebar-primary-items__end sidebar-primary__section">
+      <div class="sidebar-primary-item">
+<div id="ethical-ad-placement"
+      class="flat"
+      data-ea-publisher="readthedocs"
+      data-ea-type="readthedocs-sidebar"
+      data-ea-manual="true">
+</div></div>
  </div>
-  
-  <div id="rtd-footer-container"></div>


      </div>
@@ -847,6 +866,8 @@
        <li><a href="python/_autosummary/mlx.core.Dtype.html#mlx.core.Dtype.__init__">(Dtype method)</a>
 </li>
        <li><a href="python/_autosummary/mlx.core.DtypeCategory.html#mlx.core.DtypeCategory.__init__">(DtypeCategory method)</a>
+</li>
+        <li><a href="python/_autosummary/mlx.core.finfo.html#mlx.core.finfo.__init__">(finfo method)</a>
 </li>
        <li><a href="python/_autosummary/mlx.core.distributed.Group.html#mlx.core.distributed.Group.__init__">(Group method)</a>
 </li>
@@ -1079,7 +1100,7 @@
 </li>
      <li><a href="python/_autosummary/mlx.core.compile.html#mlx.core.compile">compile() (in module mlx.core)</a>
 </li>
-      <li><a href="cpp/ops.html#_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice">concatenate (C++ function)</a>, <a href="cpp/ops.html#_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice">[1]</a>
+      <li><a href="cpp/ops.html#_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice">concatenate (C++ function)</a>, <a href="cpp/ops.html#_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice">[1]</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.concatenate.html#mlx.core.concatenate">concatenate() (in module mlx.core)</a>
 </li>
@@ -1293,10 +1314,10 @@
 </li>
      <li><a href="cpp/ops.html#_CPPv46erfinvRK5array14StreamOrDevice">erfinv (C++ function)</a>
 </li>
-  </ul></td>
-  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="python/_autosummary/mlx.core.erfinv.html#mlx.core.erfinv">erfinv() (in module mlx.core)</a>
 </li>
+  </ul></td>
+  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="python/_autosummary/mlx.core.eval.html#mlx.core.eval">eval() (in module mlx.core)</a>

      <ul>
@@ -1320,6 +1341,12 @@
      <li><a href="python/_autosummary/mlx.core.expm1.html#mlx.core.expm1">expm1() (in module mlx.core)</a>
 </li>
      <li><a href="python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html#mlx.optimizers.exponential_decay">exponential_decay() (in module mlx.optimizers)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.export_function.html#mlx.core.export_function">export_function() (in module mlx.core)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.export_to_dot.html#mlx.core.export_to_dot">export_to_dot() (in module mlx.core)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.exporter.html#mlx.core.exporter">exporter() (in module mlx.core)</a>
 </li>
      <li><a href="cpp/ops.html#_CPPv43eyei14StreamOrDevice">eye (C++ function)</a>, <a href="cpp/ops.html#_CPPv43eyei5Dtype14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv43eyeii14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv43eyeiii14StreamOrDevice">[3]</a>, <a href="cpp/ops.html#_CPPv43eyeiii5Dtype14StreamOrDevice">[4]</a>
 </li>
@@ -1338,6 +1365,8 @@
      <li><a href="python/_autosummary/mlx.core.fft.fftn.html#mlx.core.fft.fftn">fftn() (in module mlx.core.fft)</a>
 </li>
      <li><a href="python/nn/_autosummary/mlx.nn.Module.filter_and_map.html#mlx.nn.Module.filter_and_map">filter_and_map() (Module method)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.finfo.html#mlx.core.finfo">finfo (class in mlx.core)</a>
 </li>
      <li><a href="cpp/ops.html#_CPPv47flattenRK5array14StreamOrDevice">flatten (C++ function)</a>, <a href="cpp/ops.html#_CPPv47flattenRK5arrayii14StreamOrDevice">[1]</a>
 </li>
@@ -1478,6 +1507,8 @@
      <li><a href="cpp/ops.html#_CPPv44imagRK5array14StreamOrDevice">imag (C++ function)</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.imag.html#mlx.core.imag">imag() (in module mlx.core)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.import_function.html#mlx.core.import_function">import_function() (in module mlx.core)</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.distributed.init.html#mlx.core.distributed.init">init() (in module mlx.core.distributed)</a>

@@ -1496,11 +1527,11 @@
      <li><a href="python/_autosummary/mlx.core.fft.irfft.html#mlx.core.fft.irfft">irfft() (in module mlx.core.fft)</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.fft.irfft2.html#mlx.core.fft.irfft2">irfft2() (in module mlx.core.fft)</a>
-</li>
-      <li><a href="python/_autosummary/mlx.core.fft.irfftn.html#mlx.core.fft.irfftn">irfftn() (in module mlx.core.fft)</a>
 </li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
+      <li><a href="python/_autosummary/mlx.core.fft.irfftn.html#mlx.core.fft.irfftn">irfftn() (in module mlx.core.fft)</a>
+</li>
      <li><a href="python/_autosummary/mlx.core.distributed.is_available.html#mlx.core.distributed.is_available">is_available() (in module mlx.core.distributed)</a>

      <ul>
@@ -1556,10 +1587,14 @@
 <table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
      <li><a href="python/_autosummary/mlx.core.random.key.html#mlx.core.random.key">key() (in module mlx.core.random)</a>
+</li>
+      <li><a href="python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html#mlx.nn.losses.kl_div_loss">kl_div_loss (class in mlx.nn.losses)</a>
 </li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
-      <li><a href="python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html#mlx.nn.losses.kl_div_loss">kl_div_loss (class in mlx.nn.losses)</a>
+      <li><a href="cpp/ops.html#_CPPv44kronRK5arrayRK5array14StreamOrDevice">kron (C++ function)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.kron.html#mlx.core.kron">kron() (in module mlx.core)</a>
 </li>
  </ul></td>
 </tr></table>
@@ -1865,7 +1900,7 @@
 <h2 id="P">P</h2>
 <table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
-      <li><a href="cpp/ops.html#_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice">pad (C++ function)</a>, <a href="cpp/ops.html#_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice">[3]</a>
+      <li><a href="cpp/ops.html#_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice">pad (C++ function)</a>, <a href="cpp/ops.html#_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayKNSt6stringE14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice">[3]</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.pad.html#mlx.core.pad">pad() (in module mlx.core)</a>
 </li>
@@ -1999,7 +2034,7 @@
 </li>
      <li><a href="python/nn/_autosummary/mlx.nn.RNN.html#mlx.nn.RNN">RNN (class in mlx.nn)</a>
 </li>
-      <li><a href="cpp/ops.html#_CPPv44rollRK5arrayRK5Shape14StreamOrDevice">roll (C++ function)</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayi14StreamOrDevice">[3]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayiRK5Shape14StreamOrDevice">[4]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayii14StreamOrDevice">[5]</a>
+      <li><a href="cpp/ops.html#_CPPv44rollRK5arrayRK5Shape14StreamOrDevice">roll (C++ function)</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayi14StreamOrDevice">[3]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice">[4]</a>, <a href="cpp/ops.html#_CPPv44rollRK5arrayii14StreamOrDevice">[5]</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.roll.html#mlx.core.roll">roll() (in module mlx.core)</a>
 </li>
@@ -2111,9 +2146,13 @@
 </li>
      <li><a href="python/_autosummary/mlx.core.array.size.html#mlx.core.array.size">size (array property)</a>
 </li>
-      <li><a href="cpp/ops.html#_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice">slice (C++ function)</a>, <a href="cpp/ops.html#_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice">[1]</a>
+      <li><a href="cpp/ops.html#_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice">slice (C++ function)</a>, <a href="cpp/ops.html#_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice">[3]</a>
 </li>
-      <li><a href="cpp/ops.html#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice">slice_update (C++ function)</a>, <a href="cpp/ops.html#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice">[1]</a>
+      <li><a href="python/_autosummary/mlx.core.slice.html#mlx.core.slice">slice() (in module mlx.core)</a>
+</li>
+      <li><a href="cpp/ops.html#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice">slice_update (C++ function)</a>, <a href="cpp/ops.html#_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice">[2]</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.slice_update.html#mlx.core.slice_update">slice_update() (in module mlx.core)</a>
 </li>
      <li><a href="python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html#mlx.nn.losses.smooth_l1_loss">smooth_l1_loss (class in mlx.nn.losses)</a>
 </li>
@@ -2122,11 +2161,11 @@
      <li><a href="python/nn/_autosummary/mlx.nn.Softmax.html#mlx.nn.Softmax">Softmax (class in mlx.nn)</a>
 </li>
      <li><a href="python/nn/_autosummary_functions/mlx.nn.softmax.html#mlx.nn.softmax">softmax (class in mlx.nn)</a>
-</li>
-      <li><a href="python/_autosummary/mlx.core.softmax.html#mlx.core.softmax">softmax() (in module mlx.core)</a>
 </li>
  </ul></td>
  <td style="width: 33%; vertical-align: top;"><ul>
+      <li><a href="python/_autosummary/mlx.core.softmax.html#mlx.core.softmax">softmax() (in module mlx.core)</a>
+</li>
      <li><a href="python/nn/_autosummary/mlx.nn.Softmin.html#mlx.nn.Softmin">Softmin (class in mlx.nn)</a>
 </li>
      <li><a href="python/nn/_autosummary_functions/mlx.nn.softmin.html#mlx.nn.softmin">softmin (class in mlx.nn)</a>
@@ -2145,7 +2184,7 @@
 </li>
      <li><a href="python/_autosummary/mlx.core.sort.html#mlx.core.sort">sort() (in module mlx.core)</a>
 </li>
-      <li><a href="cpp/ops.html#_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice">split (C++ function)</a>, <a href="cpp/ops.html#_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv45splitRK5arrayi14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv45splitRK5arrayii14StreamOrDevice">[3]</a>
+      <li><a href="cpp/ops.html#_CPPv45splitRK5arrayRK5Shape14StreamOrDevice">split (C++ function)</a>, <a href="cpp/ops.html#_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice">[1]</a>, <a href="cpp/ops.html#_CPPv45splitRK5arrayi14StreamOrDevice">[2]</a>, <a href="cpp/ops.html#_CPPv45splitRK5arrayii14StreamOrDevice">[3]</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.array.split.html#mlx.core.array.split">split() (array method)</a>

@@ -2337,6 +2376,10 @@
 <h2 id="U">U</h2>
 <table style="width: 100%" class="indextable genindextable"><tr>
  <td style="width: 33%; vertical-align: top;"><ul>
+      <li><a href="cpp/ops.html#_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice">unflatten (C++ function)</a>
+</li>
+      <li><a href="python/_autosummary/mlx.core.unflatten.html#mlx.core.unflatten">unflatten() (in module mlx.core)</a>
+</li>
      <li><a href="python/nn/_autosummary/mlx.nn.Module.unfreeze.html#mlx.nn.Module.unfreeze">unfreeze() (Module method)</a>
 </li>
      <li><a href="python/_autosummary/mlx.core.random.uniform.html#mlx.core.random.uniform">uniform() (in module mlx.core.random)</a>
@@ -2484,8 +2527,8 @@ By MLX Contributors
  </div>
  
  <!-- Scripts loaded after <body> so the DOM is not blocked -->
-  <script defer src="_static/scripts/bootstrap.js?digest=26a4bc78f4c0ddb94549"></script>
-<script defer src="_static/scripts/pydata-sphinx-theme.js?digest=26a4bc78f4c0ddb94549"></script>
+  <script defer src="_static/scripts/bootstrap.js?digest=8878045cc6db502f8baf"></script>
+<script defer src="_static/scripts/pydata-sphinx-theme.js?digest=8878045cc6db502f8baf"></script>

  <footer class="bd-footer">
  </footer>