Getting the stacked outputs of FixedPointIteration.run()

Dear maintainers,

In my use case I would need the output values after each update() call as well as the final fixed point value from run(). I was wondering whether it would be possible to add an option to return this stacked trace from run, rather than manually using jax.lax.scan() and update().

Best wishes,

Sophie