flax
Add proper cache behavior for lift.jit
#1275
Merged

Add proper cache behavior for lift.jit #1275

jheek
jheek Add proper cache behavior for lift.jit
9c19c13b
jheek jheek requested a review from levskaya levskaya 4 years ago
google-cla google-cla added cla: yes
jheek add type annotations
2b7582a5
levskaya
levskaya approved these changes on 2021-04-28
jheek
jheek add nonlocal to jitted function for readability
c8ecb95f
jheek jheek added pull ready
copybara-service copybara-service merged 61280ffb into master 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone