Haiku Beta Release

Changes

Examples

• Added VAE example.
• Added pruning example (https://arxiv.org/abs/1710.01878).
• MNIST example uses 300-100-10 MLP.
• Updated imagenet dataset to return correctly scaled examples.

Breaking changes

• State arg to hk.transform dropped in favor of transform_with_state.
• Decay argument is now required in BatchNorm.

Features

• Added hk.maybe_next_rng_key().
• BatchNorm and LayerNorm speed improvements.
• Added support for partition/filter/merge params.
• Haiku now allows running with jax_numpy_rank_promotion.

Experimental features

• hk.experimental.to_dot - experimental visualisation support.
• hk.experimental.lift - experimental purification support.

Usability

• Improved error message when RNG arg is not and RNG.
• Improved documentation.
• Improved test coverage.