Skip to content

Add rewrite for diag(A @ B) to avoid materializing the product#2302

Open
jessegrabowski wants to merge 2 commits into
pymc-devs:mainfrom
jessegrabowski:diag-matmul
Open

Add rewrite for diag(A @ B) to avoid materializing the product#2302
jessegrabowski wants to merge 2 commits into
pymc-devs:mainfrom
jessegrabowski:diag-matmul

Generalize the rewrite into (A @ B)[adv_idx]

cd6a3a6
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Run zizmor 🌈
succeeded Jul 19, 2026 in 15s