Removing unused description

PiperOrigin-RevId: 584384628
jax-ml · Nov 21, 2023 · db376e8 · db376e8
1 parent 84c1e82
commit db376e8
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/jax/experimental/mesh_utils.py b/jax/experimental/mesh_utils.py
@@ -137,8 +137,6 @@ def _create_device_mesh_for_nd_torus(
       physical topology.
     mesh_shape: shape of the logical mesh (size of the various logical
       parallelism axes), with axes ordered by increasing network intensity.
-    prefer_symmetric: whether to prefer to assign a logical axis to multiple
-      physical axes of the same size rather than axes of different sizes.
 
   Returns:
     An np.ndarray of devices in the shape of the logical mesh (mesh_shape), with