Bug fixes and enhancements

Improved Resnet examples in Flax and Haiku, by Fabian Pedregosa.
Documentation is built as part of the continuous integration workflow, by Fabian Pedregosa.
Jit update method, by Mathieu Blondel.
Linesearch improvements, by Vincent Roulet.
GaussNewton and LevenbergMarquardt improvements, by Amir Saadat.
Verbose support even when jit=True, by Amir Saadat.
Various doc improvements, by Fabian Pedregosa.

Contributors

Amir Saadat, Fabian Pedregosa, Mathieu Blondel, Peter Hawkins, Vincent Roulet.