What's Changed

• Fix: call apply/3 as intended by @preciz in https://github.com/elixir-nx/axon/pull/598
• Raise on ambiguous inputs by @seanmor5 in https://github.com/elixir-nx/axon/pull/599
• Doc fix for rewrite_nodes by @markevans in https://github.com/elixir-nx/axon/pull/601
• Remove dead code related to namespaces by @ahamez in https://github.com/elixir-nx/axon/pull/602
• docs: correct doc example (struct is called Node, not Nodes) by @benswift in https://github.com/elixir-nx/axon/pull/603
• Update table_rex by @chaychoong in https://github.com/elixir-nx/axon/pull/607
• Handle Model Display Correctly by @heywhy in https://github.com/elixir-nx/axon/pull/605
• fix(Issue 611): Adds Axon.Loop.checkpoint :patience option for checkpointing by @alvises in https://github.com/elixir-nx/axon/pull/610
• Update params initialization in Axon model guide by @Chapaman in https://github.com/elixir-nx/axon/pull/615
• chore: update nx by @polvalente in https://github.com/elixir-nx/axon/pull/618
• docs: add pytorch cheatsheet by @polvalente in https://github.com/elixir-nx/axon/pull/608