flax
Fix initialization RNGs to work with omnistaging for jitted inits.
#467

Merged

Fix initialization RNGs to work with omnistaging for jitted inits. #467

copybara-service merged 1 commit into master from test_332173203

google-cla added cla: yes

copybara-service force pushed 5 years ago

copybara-service force pushed 5 years ago

andsteing assigned

jheek 5 years ago

copybara-service force pushed 5 years ago

Fix initialization RNGs to work with omnistaging for jitted inits.

139d02fd

copybara-service force pushed to 139d02fd 5 years ago

copybara-service merged 139d02fd into master 5 years ago

copybara-service deleted the test_332173203 branch 5 years ago

Reviewers

No reviews

Assignees

jheek

Labels

cla: yes

Milestone

No milestone