-
Notifications
You must be signed in to change notification settings - Fork 2
Pull requests: primatrix/pallas-kernel
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
feat(gmm): JIT-compilable JAX GMM for TPU with forward/backward and tokamax comparison
#161
opened Apr 6, 2026 by
sii-xinglong
Loading…
4 tasks done
feat: multi-step forward unrolling + split backward for fused chunk GLA
#147
opened Apr 2, 2026 by
sii-xinglong
Loading…
7 of 8 tasks
feat(gla): fused forward+backward kernels for g_gamma mode
#122
opened Mar 31, 2026 by
sii-xinglong
Loading…
5 tasks done
ProTip!
Mix and match filters to narrow down what you’re looking for.