Rewrite solve(matrix_inverse(X), b) → X @ b

[alessandrogentili001]

Description

Opened a new PR and closed the previous one after getting troubles with git (my fault). By the way.

Update: Refactored Structural Testing

I have updated the PR to address the recent feedback regarding the test suite.

Key Changes:

• Idiomatic Rewrite Invocation: Switched from a custom WalkingGraphRewriter to the standard rewrite_graph(include=["stabilize"]) pattern.
• Structural Verification: Replaced numerical checks with assert_equal_computations. Both the rewritten graph and the expected X @ b graph are stabilized to ensure a common structural baseline (handling the internal lowering of Matmul to BLAS ops like Dot).
• Cleaned Up Tests: Removed manual loops and numerical validation as suggested, keeping the test suite performant and focused on the rewrite logic.

The implementation is verified to pass all tests and adheres to the latest repository linting standards.

Related Issue

• Closes Rewrite solve(matrix_inverse(X), b) → X @ b #2082
• Related to Rewrite solve(matrix_inverse(X), b) → X @ b #2101

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):

[ricardoV94]

Opened a new PR and closed the previous one after getting troubles with git (my fault). By the way.

You could have force-push to your old branch once you fixed it locally.

[ricardoV94]

Thanks @alessandrogentili001!