refactor(autogram): Make `VJP` take flat `grad_outputs`.#438

Merged

ValerianRey merged 6 commits intomainfrom

make-vjp-take-flat-grad-outputs

Oct 1, 2025

Commits on Sep 30, 2025

Make `VJP` take flat `grad_outputs`. This allows removing the parameter `output_spec` from both `autograd.Function` in `ModuleHookManager`.
PierreQuinton
committed