flax
Added keep_rngs arg to MHA to optionally store rngs
#4749

Merged

Added keep_rngs arg to MHA to optionally store rngs #4749

copybara-service merged 1 commit into google:main from vfdev-5:mha-add-keep-rngs-arg

Added keep_rngs arg to MHA to optionally store rngs

36d2421b

vfdev-5 requested a review from

cgarciae 1 year ago

cgarciae approved these changes on 2025-05-15

cgarciae added pull ready

cgarciae commented on 2025-05-15

copybara-service merged 45ce6ee2 into main 1 year ago

vfdev-5 deleted the mha-add-keep-rngs-arg branch 1 year ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone