Fix vmap axes for input_positions in dot_product_attention_with_rope - #5537
Open
TheSaiEaranti wants to merge 1 commit into
Open
Fix vmap axes for input_positions in dot_product_attention_with_rope#5537TheSaiEaranti wants to merge 1 commit into
TheSaiEaranti wants to merge 1 commit into
Google CLA / cla/google
succeeded
Jul 26, 2026 in 7s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
The following contributors were found for this pull request:
✅ f2f0bb7 Author: @TheSaiEaranti <saipr*********anti@utexas.edu>
(Only the first commit for a unique contributor is listed.)
Loading