Use the recommended way to set RBG PRNG. Allow us to jnp.stack on a list of RBG PRNGKeys today.
Use the recommended way to set RBG PRNG. Allow us to jnp.stack on a list of RBG PRNGKeys today.

PiperOrigin-RevId: 429215384

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo