Open miltonllera opened 2 months ago
All calls to jax.random.PRNGKey have been changed to jax.random.key. When passing keys to non-jax methods (such as the ones in scikit-learn), we first jax.key_data to recover the underlying raw key information before making the call.
jax.random.PRNGKey
jax.random.key
jax.key_data
Related issues: https://jax.readthedocs.io/en/latest/jep/9263-typed-keys.html
All calls to
jax.random.PRNGKey
have been changed tojax.random.key
. When passing keys to non-jax methods (such as the ones in scikit-learn), we firstjax.key_data
to recover the underlying raw key information before making the call.Related issues: https://jax.readthedocs.io/en/latest/jep/9263-typed-keys.html