flax
[linen] fold rngs on jit to improve caching
#4064
Merged

[linen] fold rngs on jit to improve caching #4064

copybara-service merged 1 commit into main from test_649462075
copybara-service
copybara-service copybara-service force pushed from 48918472 to 72098921 1 year ago
copybara-service copybara-service force pushed from 72098921 1 year ago
copybara-service copybara-service force pushed 1 year ago
copybara-service copybara-service force pushed 1 year ago
[linen] fold rngs on jit to improve caching
afcaf66d
copybara-service copybara-service force pushed to afcaf66d 1 year ago
copybara-service copybara-service merged afcaf66d into main 1 year ago
copybara-service copybara-service deleted the test_649462075 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone