Fix vmap of scatter placing the batch axis wrongly in the updates - #4322
Open
Adityaj0 wants to merge 1 commit into
Open
Fix vmap of scatter placing the batch axis wrongly in the updates#4322Adityaj0 wants to merge 1 commit into
Adityaj0 wants to merge 1 commit into