Skip to content

Pull requests: aws-neuron/nki-samples

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
Assignee
Filter by who’s assigned
Sort

Pull requests list

Disable DMA transpose on attention forward kernel
#42 opened Jan 9, 2025 by aws-zhehongb Loading… updated Jan 20, 2025
Implement the LoRA forward and backward passes with NKI
#46 opened Jan 21, 2025 by nandeeka Loading… updated Jan 21, 2025
5 tasks done
Adding advanced FAQ
#47 opened Jan 27, 2025 by EmilyWebber Loading… updated Jan 30, 2025
6 tasks done
[Bug Fix] there should be no addition of result of the first matmul t…
#50 opened Feb 14, 2025 by akothen Loading… updated Feb 14, 2025
Update contributed matmul to be compatible with Neuron version bump
#45 opened Jan 21, 2025 by nandeeka Loading… updated Feb 22, 2025
5 tasks done
Rotary Embedding NKI Kernel Tutorial
#53 opened Feb 27, 2025 by apatwal72 Loading… updated Feb 27, 2025
3 of 6 tasks
Double row matmul
#49 opened Feb 11, 2025 by aws-maxkoh Loading… updated Feb 27, 2025
5 of 6 tasks
feat: add softmax examples
#55 opened Mar 3, 2025 by jlonge4 Loading… updated Mar 5, 2025
4 of 6 tasks
ProTip! no:milestone will show everything without a milestone.