Add update reporting to Monitor.update()
Merge remote-tracking branch 'localflax/1.1.x' into topic/aw-parallel