[JAX] Grouped GEMM Refactor to use first_dims and last_dims#2749
Draft
jberchtold-nvidia wants to merge 12 commits intoNVIDIA:mainfrom
Draft
[JAX] Grouped GEMM Refactor to use first_dims and last_dims#2749jberchtold-nvidia wants to merge 12 commits intoNVIDIA:mainfrom
jberchtold-nvidia wants to merge 12 commits intoNVIDIA:mainfrom
Commits
Commits on Mar 9, 2026
Commits on Mar 10, 2026
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}