chex
chex copied to clipboard
Update PRNGKey type in accordance with https://github.com/google/jax/issues/16716.
Update PRNGKey type in accordance with https://github.com/google/jax/issues/16716.
This actually breaks chex now. Would it be possible to merge this PR and make a release, please?