flax
use more efficient softmax in dot product attention.
#334
Merged

use more efficient softmax in dot product attention. #334

copybara-service merged 1 commit into master from test_317835753
copybara-service
googlebot googlebot added cla: yes
codecov-commenter
avital avital assigned levskaya levskaya 5 years ago
avital
levskaya
copybara-service copybara-service force pushed 5 years ago
levskaya use more efficient softmax in dot product attention.
7fcc7f6c
copybara-service copybara-service force pushed to 7fcc7f6c 5 years ago
copybara-service copybara-service closed this 5 years ago
copybara-service copybara-service merged 7fcc7f6c into master 5 years ago
copybara-service copybara-service deleted the test_317835753 branch 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
Labels
Milestone