DOC: jax.lax.top_k: fix rendering of return values

Author: jakevdp

jax/_src/lax/lax.py

@@ -1232,8 +1232,10 @@ def top_k(operand: ArrayLike, k: int) -> tuple[Array, Array]:
     k: integer specifying the number of top entries.
 
   Returns:
-    values: array containing the top k values along the last axis.
-    indices: array containing the indices corresponding to values.
+    A tuple ``(values, indices)`` where
+
+    - ``values`` is an array containing the top k values along the last axis.
+    - ``indices`` is an array containing the indices corresponding to values.
 
   See also:
   - :func:`jax.lax.approx_max_k`