Skip to content

Commit f2d2d71

Browse files
author
jax authors
committed
Merge pull request #20136 from yixiaoer:fix-typo
PiperOrigin-RevId: 613961269
2 parents 4244b21 + 6ada248 commit f2d2d71

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

jax/random.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,8 +116,9 @@
116116
identical across JAX/XLA versions ✅ ✅
117117
================================= ======== ========= === ========== ===== ============
118118
119-
(*): with jax_threefry_partitionable=1 set
120-
(**): with XLA_FLAGS=--xla_tpu_spmd_rng_bit_generator_unsafe=1 set
119+
(*): with ``jax_threefry_partitionable=1`` set
120+
121+
(**): with ``XLA_FLAGS=--xla_tpu_spmd_rng_bit_generator_unsafe=1`` set
121122
122123
The difference between "rbg" and "unsafe_rbg" is that while "rbg" uses a less
123124
robust/studied hash function for random value generation (but not for

0 commit comments

Comments
 (0)