[JAX] Support for batched einsum and grouped GEMM without D2H memcpy#2604
Draft
jberchtold-nvidia wants to merge 44 commits intoNVIDIA:mainfrom
Draft
[JAX] Support for batched einsum and grouped GEMM without D2H memcpy#2604jberchtold-nvidia wants to merge 44 commits intoNVIDIA:mainfrom
jberchtold-nvidia wants to merge 44 commits intoNVIDIA:mainfrom
Commits
Commits on Dec 3, 2025
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}
Commits on Dec 10, 2025
- committed
- committed
- committed
- committed
- committed
- committed
Commits on Dec 11, 2025
- committed
- committed
- committed
- committed
- committed
Commits on Dec 16, 2025
Commits on Dec 17, 2025
- committed
Commits on Dec 18, 2025
- committed
Commits on Dec 19, 2025
- committed
- authored
- committed
- committed
- committed
Commits on Dec 22, 2025
- committed
- committed
- committed
- committed
Commits on Dec 23, 2025
- committed
- committed
- committed
- committed
- committed
- committed
- committed
Commits on Jan 5, 2026
- committed
Commits on Jan 7, 2026
Commits on Jan 8, 2026
- committed
- committed
- committed
- committed
Commits on Jan 15, 2026
- committed
- committed
- committed