Clarify more the potential performance improvements in `optimize_rng_use`.
Clarify more the potential performance improvements in `optimize_rng_use`.

PiperOrigin-RevId: 447709840

deepmind/dm-haiku,deepmind/dm-haiku