Fix useless unbatched Blockwise rewrite

[N-0-MAD]

Description

This fixes an issue in local_useless_unbatched_blockwise where removing dummy batch dimensions could leave behind redundant Squeeze(ExpandDims(x)) expressions.

The rewrite now applies apply_local_dimshuffle_lift after squeezing the dummy dimensions, allowing these redundant dimension operations to be simplified before subsequent rewrites.

This prevents equivalent computations from being treated as distinct and allows common factorizations to be merged. A regression test was added demonstrating that equivalent Cholesky factorizations result in a single Cholesky node.

Related Issue

• Closes local_useless_unbatched_blockwise emits squeeze(expand_dims(x)) that nothing collapses, so factorizations stop merging #2416
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):