Fixed printing order of results in jax.debug.print documentation.

jax-ml · Feb 28, 2025 · 6caf447 · 6caf447
1 parent d8953e5
commit 6caf447
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/debugging/print_breakpoint.md b/docs/debugging/print_breakpoint.md
@@ -91,8 +91,8 @@ def f(x):
   jax.debug.print("x: {}", x)
   return x
 jax.pmap(f)(xs)
-# Prints: x: 1.0
-#         x: 0.0
+# Prints: x: 0.0
+#         x: 1.0
 # OR
 # Prints: x: 1.0
 #         x: 0.0