-
Notifications
You must be signed in to change notification settings - Fork 131
Pull requests: elixir-nx/axon
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Fix sparse categorical cross entropy for softmax outputs
#680
opened Sep 17, 2026 by
dajiaohuang
Loading…
Keep metric accumulators and transforms during validation
#679
opened Sep 17, 2026 by
georgeguimaraes
Member
Loading…
Document how initializers read custom parameter shapes
#678
opened Sep 17, 2026 by
georgeguimaraes
Member
Loading…
Let a loss function take the batch inputs
#676
opened Sep 17, 2026 by
georgeguimaraes
Member
Loading…
docs: Say that dense works on inputs of any rank
#674
opened Sep 15, 2026 by
georgeguimaraes
Member
Loading…
chore: Update to Nx, EXLA and Torchx 1.0
#671
opened Sep 15, 2026 by
georgeguimaraes
Member
Loading…
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…
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…
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!
Type g p on any issue or pull request to go back to the pull request listing page.