flax
Fix lifting rng splits
#1531
Merged

Fix lifting rng splits #1531

jheek
jheek Dynamically check whether named call is enabled.
6c46a167
jheek Remove unnecessary splitting of rngs in lifting logic.
0d69f1a2
jheek jheek requested a review from levskaya levskaya 4 years ago
google-cla google-cla added cla: yes
levskaya
levskaya approved these changes on 2021-09-09
levskaya levskaya added pull ready
copybara-service copybara-service merged cf638d18 into main 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone