Refactor and improve `Pad` - #2315

Open

jessegrabowski wants to merge 10 commits into

pymc-devs:mainfrom

jessegrabowski:symbolic-pad

Commits on Jul 30, 2026

Reimplement wrap/symmetric/reflect padding as index gathers
jessegrabowski
committed
3f339a8
Add xfail tests for pad static shapes and gradient jax-compatibility
jessegrabowski
committed
00acb58
Add xfail test for broken edge-mode pad gradient
jessegrabowski
committed
317085f
Report an invalid pad mode as ValueError, not KeyError
jessegrabowski
committed
828a1ad
Convert Pad from OpFromGraph to SymbolicOp
jessegrabowski
committed
3bb9e9c
Fix edge-mode pad gradient by pinning the edge width to 1
jessegrabowski
committed
756da08