flax
fix omnistaging issues
#335
Merged

fix omnistaging issues #335

jheek
jheek fix omnistaging issues
1f553886
googlebot googlebot added cla: yes
jheek jheek requested a review from levskaya levskaya 5 years ago
jheek use staging jaxpr trace in lazy init
6ccd6a60
jheek fix howto
ee20ca22
codecov-commenter
levskaya
levskaya approved these changes on 2020-06-26
copybara-service copybara-service merged daff4030 into master 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone