Skip to content

Pull requests: aws-neuron/nki-samples

New pull request New
Clear current search query, filters, and sorts
8 Open 10 Closed
8 Open 10 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

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