Skip to content

Pull requests: elixir-nx/axon

New pull request New
13 Open 395 Closed
13 Open 395 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

Add training loop fusion
#658 opened Aug 16, 2026 by seanmor5 Contributor Loading…
1
Remove loop output transforms
#657 opened Aug 16, 2026 by seanmor5 Contributor Loading…
1
Define Axon.ModelState functions as transforms
#656 opened Aug 15, 2026 by seanmor5 Contributor Loading…
1
Add Axon.Loop.stream/4 and implement run/4 on top of it
#655 opened Aug 15, 2026 by seanmor5 Contributor Loading…
Fix quantization metadata loss and non-f32 weight quantization
#654 opened Aug 15, 2026 by seanmor5 Contributor Loading…
Add Axon.elem/3 and Axon.fetch/3 container accessors
#653 opened Aug 15, 2026 by seanmor5 Contributor Loading…
Add Axon.namespace/2 and /3
#652 opened Aug 15, 2026 by seanmor5 Contributor Loading…
Add Axon.deferred/2
#651 opened Aug 15, 2026 by seanmor5 Contributor Loading…
Add scaled dot-product attention
#650 opened Aug 15, 2026 by seanmor5 Contributor Loading…
1
Fix RNN unrolling gradients and bidirectional LSTMs
#646 opened Aug 15, 2026 by seanmor5 Contributor Loading…
1
Donate step state buffers during training
#645 opened Aug 15, 2026 by seanmor5 Contributor Loading…
Add Axon.capture
#643 opened Aug 14, 2026 by Chapaman Contributor • Draft
Add Axon.capture
#626 opened Jan 31, 2026 by seanmor5 Contributor Loading…
8
ProTip! Filter pull requests by the default branch with base:main.