Haiku 0.0.13

tomhennigan released this 16 Oct 09:11

3009ec9

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).

Assets 2