Update ad.backward_pass to support non-linear functions of constants#26811

Open

dfm wants to merge 1 commit intojax-ml:mainfrom dfm:direct-lin

+34-1

Commits on Feb 27, 2025

Update ad.backward_pass to support non-linear functions of constants.
dfm
committed