Skip to content

Extend inv(A) @ b -> solve(A, b) to batched matmul#2157

Merged
ricardoV94 merged 1 commit into
pymc-devs:mainfrom
ricardoV94:batch_inv_to_solve
May 20, 2026
Merged

Extend inv(A) @ b -> solve(A, b) to batched matmul#2157
ricardoV94 merged 1 commit into
pymc-devs:mainfrom
ricardoV94:batch_inv_to_solve

Conversation

@ricardoV94
Copy link
Copy Markdown
Member

@ricardoV94 ricardoV94 commented May 20, 2026

Closes #1895
Closes #1920
Closes #945

@ricardoV94 ricardoV94 changed the title Extend Inv to Solve to batched matmul Extend inv(A) @ b -> solve(A, b) to batched matmul May 20, 2026
@ricardoV94
Copy link
Copy Markdown
Member Author

Benchmark failure is due to github actions being merged out of sync by the time this ran, should be fixed in main already and unrelated

@ricardoV94 ricardoV94 merged commit 6619bef into pymc-devs:main May 20, 2026
65 of 66 checks passed
@ricardoV94 ricardoV94 deleted the batch_inv_to_solve branch May 20, 2026 15:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Extend inverse @ b as solve rewrite to batched dots

2 participants