KevGomes1403 / nki-moe-megakernelView on GitHub
Fused multilayer transformer kernels on Trainium 2/3
☆11Aug 17, 2026Updated 2 months ago

Alternatives and similar repositories for nki-moe-megakernel

Users that are interested in nki-moe-megakernel are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?