-
Notifications
You must be signed in to change notification settings - Fork 104
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.
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
Provide a better error message on shape mismatches in training loops
#480
opened Feb 27, 2023 by
seanmor5
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
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
ProTip!
Follow long discussions with comments:>50.