Skip to content

Issues: elixir-nx/axon

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

17 Open 223 Closed
17 Open 223 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
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Construct train step from an objective function and optimizer
#595 opened Sep 11, 2024 by seanmor5
Crash with mixed precision with new Nx
#578 opened May 30, 2024 by seanmor5
Optionally return gradients / gradient state in train step
#577 opened May 15, 2024 by seanmor5
Merge model state updates into a single Axon.ModelState call
#576 opened May 15, 2024 by seanmor5
Most Axon.ModelState functions should be defn or deftransform
#575 opened May 15, 2024 by seanmor5
Compile Error due to a type mismatch
#568 opened Apr 20, 2024 by Mostafa86
2
Add a guide for writing a simple recurrent network
#511 opened Jul 10, 2023 by marcinkoziej
7
Provide a better error message on shape mismatches in training loops
#480 opened Feb 27, 2023 by seanmor5
Experiences from porting YOLOv8 to Axon
#475 opened Feb 12, 2023 by hansihe
10
Consider making all loop executions run in a separate process
#455 opened Jan 19, 2023 by seanmor5
1
Support gradient checkpoints in models kind:feature New feature or request note:upstream The issue must be tackled upstream
#372 opened Sep 20, 2022 by seanmor5
Add multi-device training/inference kind:feature New feature or request note:upstream The issue must be tackled upstream
#371 opened Sep 20, 2022 by seanmor5
Default loop output transforms are too intrusive area:loop Applies to loop API kind:chore Internal Improvements
#362 opened Sep 17, 2022 by seanmor5
2
1
Add model pruning kind:feature New feature or request
#120 opened Sep 24, 2021 by seanmor5
Add parameter constraints area:optimizers Applies to optimization API kind:feature New feature or request
#111 opened Sep 13, 2021 by seanmor5
Add training loop fusion area:loop Applies to loop API kind:feature New feature or request
#101 opened Aug 27, 2021 by seanmor5
Integrate named tensors kind:feature New feature or request
#57 opened Apr 19, 2021 by seanmor5
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.