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 6 years ago

copybara-service force pushed 6 years ago

andsteing assigned

jheek 6 years ago

copybara-service force pushed 6 years ago

Fix initialization RNGs to work with omnistaging for jitted inits.

139d02fd

copybara-service force pushed to 139d02fd 6 years ago

copybara-service merged 139d02fd into master 6 years ago

copybara-service deleted the test_332173203 branch 6 years ago

Reviewers

No reviews

Assignees

jheek

Labels

cla: yes

Milestone

No milestone