fix: return the polished raw-forward PDIP iterate as the forward value (#594) - #595

Merged

Jammy2211 merged 3 commits into

mainfrom

feature/raw-pdip-forward-polish

Sep 30, 2026

Commits on Sep 30, 2026

test: red amplitude regression for the raw-forward PDIP forward value (#594)

authored and
Jammy2211
committed
7e62fa4
fix: return the polished raw-forward PDIP iterate as the forward value (#594)

authored and
Jammy2211
committed
31b1c2d
test: end-to-end jit-vs-eager check tolerates runner-dependent XLA reassociation (#594)

authored and
Jammy2211
committed
42c5235