-
Notifications
You must be signed in to change notification settings - Fork 129
Pull requests: elixir-nx/axon
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Build train steps from objective functions
#669
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Initialize training step state in the types the step produces
#668
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Add Axon.Pruning with magnitude-based unstructured pruning
#666
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Add axis names to Axon.input and an Axon.rename layer
#664
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Merge training updates into a single Axon.ModelState.update call
#663
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Improve shape mismatch errors in training loops
#662
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Optionally keep gradients in the training step state
#661
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Diagnose partial initial model states and document Axon.build
#660
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
Log iteration instead of batch in Axon.Loop progress output
#659
opened Aug 23, 2026 by
seanmor5
Contributor
Loading…
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…
ProTip!
What’s not been updated in a month: updated:<2026-07-24.