New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

New jax keys #703

Closed

reubenharry opened this issue Jun 24, 2024 · 1 comment

Contributor

reubenharry commented Jun 24, 2024

Current behavior

We currently use jax.random.PRNGKey throughout.

Desired behavior

Following this jax discussion https://jax.readthedocs.io/en/latest/jep/9263-typed-keys.html, I'm wondering if we want to switch to the new key style, jax.random.key

Member

junpenglao commented Jun 25, 2024

We are already using the new random key: #569. It is just we still call the custom type PRNGKey

junpenglao closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment