flax
Allow using LazyRngs for flax init/apply.
#4818
Merged

Allow using LazyRngs for flax init/apply. #4818

copybara-service merged 1 commit into main from test_783557644
copybara-service
copybara-service copybara-service force pushed from da766c89 to 01394e52 1 year ago
copybara-service copybara-service force pushed from 01394e52 to 50c899a0 1 year ago
levskaya Allow using LazyRngs for flax init/apply.
1ccd1598
copybara-service copybara-service force pushed from 50c899a0 to 1ccd1598 1 year ago
copybara-service copybara-service merged 1ccd1598 into main 1 year ago
copybara-service copybara-service deleted the test_783557644 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone