v0.4.0

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_table by @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