[JAX] Grouped quant+GEMM custom partitioning rules#3058
Draft
jberchtold-nvidia wants to merge 4 commits into
Draft
[JAX] Grouped quant+GEMM custom partitioning rules#3058jberchtold-nvidia wants to merge 4 commits into
jberchtold-nvidia wants to merge 4 commits into
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}