dev

[patrick-kidger]

• Fixed {Newton,Chord}(cauchy_termination=False) failing when started close to the solution
• Fixed sol.state including only the dynamic part
• Fixed test function unwrapping with latest jaxtyping
• Moved norms to Lineax
• Updated to use eqxi.GetKey and tree_allclose
• Switch to ruff-format and ruff for ipynb
• Fixed implicit_jvp assuming that it was only being used with iterate
• Now using the same jaxpr in the state.