jax

a486f548 - Add a summary explaining the usage and context for JAX PRNG design. (#2525)

Add a summary explaining the usage and context for JAX PRNG design. (#2525) * Add a summary explaining the usage and context for JAX PRNG design. The current design_notes do not match current JAX API, and it's a pretty long doc to read to understand how to use it. Closes: #2087 * Change 'should' to be more precise. * Address comments.