flax
[nnx] add Dropout.rngs
#3815
Merged

[nnx] add Dropout.rngs #3815

cgarciae
cgarciae [nnx] add Dropout.rngs
1a1ca486
cgarciae cgarciae changed the base branch from nnx-rng-tags to main 2 years ago
chiamp
chiamp approved these changes on 2024-04-02
chiamp chiamp added pull ready
copybara-service copybara-service merged 27184559 into main 2 years ago
copybara-service copybara-service deleted the nnx-dropout-optional-state branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone