Skip to content

Haiku 0.0.10

Compare
Choose a tag to compare
@tomhennigan tomhennigan released this 14 Jul 15:42
  • Added hk.mixed_precision.push_policy.
  • Added hk.experimental.{get_params,get_initial_state,get_current_state}.
  • Added hk.experimental.{maybe_get_rng_sequence_state,maybe_replace_rng_sequence_state}.
  • hk.switch now supports multiple operands.
  • hk.get_parameter now supports init=None.
  • hk.MethodContext now includes orig_class.
  • hk.GetterContext now includes lifted_prefix_name.
  • hk.layer_stack now allows parameter reuse.
  • Haiku is now compatible with jax.enable_custom_prng.
  • TruncatedNormal now exports lower and upper bounds.
  • Haiku init/apply functions now return dict rather than Mapping.
  • hk.dropout now supports broadcast_dims.