WIP: Add rewrite to fuse nested BlockDiag Ops#1671

Draft

eby0303 wants to merge 3 commits intopymc-devs:mainfrom

eby0303:fuse-blockdiag-rewrite

Commits on Oct 16, 2025

Commits on Nov 4, 2025

linalg: fuse nested BlockDiagonal ops and add corresponding tests
eby0303
committed