Preserve static shape information in block_diag
#1529
Merged
jessegrabowski merged 1 commit intopymc-devs:mainfrom Jul 9, 2025
Merged
Preserve static shape information in `block_diag`#1529jessegrabowski merged 1 commit intopymc-devs:mainfrom
jessegrabowski merged 1 commit intopymc-devs:mainfrom