flax
Add functool.wraps() annotation to flax.nn.jit.
#4051
Merged

Add functool.wraps() annotation to flax.nn.jit. #4051

copybara-service merged 1 commit into main from test_648751795
copybara-service
copybara-service copybara-service force pushed 2 years ago
hawkinsp Add functool.wraps() annotation to flax.nn.jit.
37123d52
copybara-service copybara-service force pushed to 37123d52 2 years ago
copybara-service copybara-service merged 37123d52 into main 2 years ago
copybara-service copybara-service deleted the test_648751795 branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone