flax
MultiHeadAttention only keeps rngs if dropout_rate is positive
#4750

Merged

MultiHeadAttention only keeps rngs if dropout_rate is positive #4750

copybara-service merged 1 commit into main from test_759381255

copybara-service force pushed from 99a6e373 to ce70c939 223 days ago

MultiHeadAttention only keeps rngs if dropout_rate is positive

65dfd465

copybara-service force pushed from ce70c939 to 65dfd465 223 days ago

copybara-service merged 65dfd465 into main 223 days ago

copybara-service deleted the test_759381255 branch 223 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone