Merge branch 'main' into dependabot/github_actions/actions/checkout-4…

….2.2

doc/requirements.txt

@@ -1,5 +1,5 @@
 Sphinx==8.1.3
-sphinx-autoapi===3.3.2
+sphinx-autoapi===3.3.3
 sphinx_rtd_theme==3.0.1
 sphinxcontrib-napoleon==0.7
 sphinx-copybutton==0.5.2

heat/core/manipulations.py

@@ -3719,7 +3719,8 @@ def _axis2axisResplit(
     comm: Communication,
 ) -> torch.Tensor:
     """
-    Resplits the input array along a new axis and performs data exchange using MPI_Alltoallw. Returns target_larray object with the data after the exchange.
+    Resplits the input array along a new axis and performs data exchange using MPI_Alltoallw, after [1].
+    Returns `target_larray` object with the data after the exchange.
 
     Parameters
     ----------
@@ -3737,6 +3738,10 @@ def _axis2axisResplit(
         The tiling object containing the subarray parameters for the target array.
     comm : Communication
         The communication object for MPI communication.
+
+    References
+    ----------
+    [1] Dalcin, Mortensen, Keyes, "Fast parallel multidimensional FFT using advanced MPI", 2018.
     """
     # Create subarray types for original local shapes split along the new axis
     source_subarray_params = source_tiles.get_subarray_params(source_split, target_split)