flax
Validate dropout_rate in dot_product_attention
#5503

Merged

Validate dropout_rate in dot_product_attention #5503

copybara-service merged 2 commits into google:main from mohsinm-dev:validate-dropout-rate

vfdev-5 commented on 2026-06-17

validate dropout_rate in dot_product_attention

89b5e6b8

mohsinm-dev force pushed from 66934c02 to 89b5e6b8 56 days ago

merge main into validate-dropout-rate

8f75c1cc

samanklesaria requested a review from

samanklesaria 38 days ago

samanklesaria approved these changes on 2026-07-06

IvyZX approved these changes on 2026-07-21

IvyZX added pull ready

copybara-service merged 7537c605 into main 21 days ago

Reviewers

IvyZX

samanklesaria

vfdev-5

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone