Haiku 0.0.13

• Various changes to remain compatible with latest JAX release (e.g. #779, #783, #789).
• Add shard_map flops estimate to jaxpr_info (#791).
• Correct an oversight in TruncatedNormal's documentation (#793).