What's Changed
- Add gradient accumulation by @seanmor5 in #397
- Update notebook dependencies by @tcoyze in #398
- Return init params on compile by @josevalim in #400
- Update your_first_axon_model.livemd by @joelpaulkoch in #401
- Remove redundant cells by @msluszniak in #402
- Support lazy container inputs by @seanmor5 in #403
- Start an integration test suite by @seanmor5 in #404
- Unify event handlers in loops by @seanmor5 in #405
- Doc latex fixes by @chadykamar in #406
- Fix issue with initialization missing keys for some layers by @seanmor5 in #407
- Fix issue with namespaced stateful models by @seanmor5 in #409
- Training improvements by @seanmor5 in #411
- Add plot handler by @cigrainger in #412
- Training upgrades by @seanmor5 in #420
- Update schedules to require initial value by @seanmor5 in #421
- Make training deterministic by @seanmor5 in #424
- Use global event counts for correct filters in loops by @seanmor5 in #425
- Remove cast to Nx.Defn.Expr and simplify logits in losses by @josevalim in #429
- Build evaluator and zero metrics only once by @josevalim in #428
- Use a layer state to manage dropout state so it changes between runs by @seanmor5 in #426
- Provide deterministic builds by @josevalim in #427
- Use a more assertive name on plotting by @josevalim in #418
- Fix conv transpose with channels last by @seanmor5 in #431
- Fix Basic Examples by @nskins in #437
- Implement dropout as a passthrough by @josevalim in #436
- Only compute random key on init by @josevalim in #435
- Remove extra usage of 'expected input' from error message in get_input/3 by @nskins in #438
- Fix text generator example by @nskins in #441
- Add paragraph on integrating with Nx.Serving by @jonatanklosko in #442
- Include the folder for the guide links by @matthusby in #445
- Fix typo in
Axon.Display.as_tableby @NduatiK in #447 - Fix documentation for evaluator by @seanmor5 in #449
- Fix Fashion MNIST autoencoder example by @nskins in #439
- Fix issues with mismatched templates by @seanmor5 in #450
- Fix deprecation warnings with new Nx by @seanmor5 in #454
- Updates for release 0.4 by @seanmor5 in #456
New Contributors
- @tcoyze made their first contribution in #398
- @joelpaulkoch made their first contribution in #401
- @chadykamar made their first contribution in #406
- @matthusby made their first contribution in #445
Full Changelog: v0.3.0...v0.4.0