What's the different between jax.random.PRNGKey and jax.random.key?

[hilanzy]

Hi, I found there is a new api jax.random.key has been add in the latest version(I use 0.4.9 berfore), what's the different between jax.random.PRNGKey and jax.random.key? Will jax.random.PRNGKey be deprecated in the future version?

[jakevdp]

In short, jax.random.key produces new-style typed keys, jax.random.PRNGKey produes older-style raw keys, and will eventually be deprecated. For more, see https://jax.readthedocs.io/en/latest/jep/9263-typed-keys.html

[hilanzy]

Thanks for your information.