PR #3354 use jax.Array type for rng keys

use jax.Array type for rng keys #3354

copybara-service merged 1 commit into google:main from chiamp:dunder

chiamp assigned

chiamp 2 years ago

chiamp force pushed 2 years ago

use jax.Array type for rng keys

dbc92549

chiamp force pushed to dbc92549 2 years ago

chiamp requested a review from

IvyZX 2 years ago

IvyZX approved these changes on 2023-09-21

IvyZX added pull ready

copybara-service merged d33a33e6 into main 2 years ago

chiamp deleted the dunder branch 2 years ago

Reviewers

IvyZX

Assignees

chiamp

Labels

pull ready

Milestone

No milestone