flax
Allow manually specifying the rng key for Dropout
#3114

Merged

Allow manually specifying the rng key for Dropout #3114

copybara-service merged 1 commit into main from explicit-dropout-rng

allow manually specifying the rng key for Dropout

4bed5e92

cgarciae force pushed to 4bed5e92 2 years ago

cgarciae commented on 2023-05-24

cgarciae requested a review from

jheek 2 years ago

IvyZX approved these changes on 2023-05-24

IvyZX added pull ready

copybara-service merged b87f5b32 into main 2 years ago

copybara-service deleted the explicit-dropout-rng branch 2 years ago

Reviewers

IvyZX

jheek

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone