Validate dropout_rate in dot_product_attention#5503

mohsinm-dev wants to merge 2 commits into

google:maingoogle/flax:mainfrom

mohsinm-dev:validate-dropout-ratemohsinm-dev/flax:validate-dropout-rateCopy head branch name to clipboard