DNM [vjp3] Make value_and_grad always use f_vjp.with_refs(*accums)#32470

gspschmid wants to merge 1 commit intojax-ml:mainjax-ml/jax:mainfrom

gspschmid:gschmid/vjp3-withrefs-testgspschmid/jax:gschmid/vjp3-withrefs-testCopy head branch name to clipboard