fix: Change to new-style-jax-rng-keys. #195
Merged
Commits on May 21, 2024
-
fix: Change to new-style-jax-rng-keys.
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 use jax.key_data to recover the underlying raw key information.
Commits on Sep 13, 2024
-
Merge branch 'new-style-jax-rng-keys' of github.com:miltonllera/QDax …
…into miltonllera-new-style-jax-rng-keys
-
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.