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

2025-04-14 10:56:06 +00:00 · 2025-02-28 06:17:14 +00:00 · 2025-02-28 06:17:14 +00:00 · 56285aec6b
commit 56285aec6b
parent d8953e5311
1 changed files with 2 additions and 2 deletions
--- 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