Fix missing state in training (#579) · elixir-nx/axon@57cd12f

Node.js 16 actions are deprecated. Please update the following actions to use Node.js 20: actions/checkout@v3, actions/cache@v3, JamesIves/[email protected]. For more information see: https://github.blog/changelog/2023-09-22-github-actions-transitioning-from-node-16-to-node-20/.

The `set-output` command is deprecated and will be disabled soon. Please upgrade to using Environment Files. For more information see: https://github.blog/changelog/2022-10-11-github-actions-deprecating-save-state-and-set-output-commands/

documentation references function "Axon.Layers.conv_lstm_cell/5" but it is undefined or private

documentation references function "Nx.Defn.jit/3" but it is undefined or private

documentation references function "Axon.Loop.log/4" but it is undefined or private

documentation references function "Axon.Loop.handle/4" but it is hidden

documentation references function "Axon.Layers.conv_lstm_cell/5" but it is undefined or private

documentation references function "Nx.Defn.jit/3" but it is undefined or private

documentation references function "Axon.Loop.log/4" but it is undefined or private

documentation references function "Axon.Loop.handle/4" but it is hidden

Provide feedback