Haiku 0.0.7
- Bug fix: modules with leading zeros (e.g.
linear_007
) are now correctly handled. 7632aff - Breaking change:
hk.vmap(..)
now requiressplit_rng
to be passed. - Breaking change:
hk.jit
was removed from the public API. - Feature: we always add profiler name scopes to Haiku modules with the latest version of JAX.
- Added a tutorial on parameter sharing.
- Added
hk.ModuleProtocol
andhk.SupportsCall
. - Added
cross_replica_axis
toVectorQuantiser
. - Added
allow_reuse
argument tohk.lift
. - Added
fan_in_axes
toVarianceScaling
initialiser. - Added
hk.custom_setter(..)
to intercepthk.set_state(..)
. - Added
hk.Deferred
. - Added
hk.experimental.transparent_lift(..)
andhk.experimental.transparent_lift_with_state(..)
. - Added
hk.experimental.fast_eval_shape(..)
. - Added
hk.experimental.current_name()
. - Added
hk.experimental.DO_NOT_STORE
. 2a6c034 - Added config APIs.
- Added support for new
jax.named_call
implementation. - The
HAIKU_FLATMAPPING
env var is no longer used. hk.dropout(..)
now supports dynamicrate
.hk.without_apply_rng(..)
now supports multi transformed functions.