Skip to content

Pull requests: primatrix/pallas-kernel

New pull request New
8 Open 142 Closed
8 Open 142 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[Feat] Add implement for kda_chunk_fwd beaver/needs-split cpu-ref Modifies tops/cpu/ reference implementations
#193 opened Apr 13, 2026 by lingebeng Collaborator • Draft
5 of 12 tasks
8
feat: add fully-fused Pallas chunk KDA forward kernel beaver/needs-split
#192 opened Apr 13, 2026 by FENP Contributor • Draft
5 tasks done
8
feat: KDA Pallas intra-chunk kernel beaver/needs-split
#188 opened Apr 13, 2026 by 0xaskr Collaborator Loading…
3 of 4 tasks
29
feat: FP8 blockwise-quantized GMM/TGMM Pallas kernels
#184 opened Apr 9, 2026 by pengchengneo Loading…
1 of 2 tasks
6
feat(kda): add KDA Pallas kernel with chunked forward + backward beaver/needs-split
#174 opened Apr 7, 2026 by 0xaskr Collaborator • Draft
3 tasks done
18
feat(gmm): JIT-compilable JAX GMM for TPU with forward/backward and tokamax comparison beaver/stale
#161 opened Apr 6, 2026 by sii-xinglong Contributor Loading…
4 tasks done
8
feat: multi-step forward unrolling + split backward for fused chunk GLA beaver/stale
#147 opened Apr 2, 2026 by sii-xinglong Contributor Loading…
7 of 8 tasks
4
feat(gla): fused forward+backward kernels for g_gamma mode beaver/stale
#122 opened Mar 31, 2026 by sii-xinglong Contributor Loading…
5 tasks done
15
ProTip! Type g i on any issue or pull request to go back to the issue listing page.