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_rngto be passed. - Breaking change:
hk.jitwas 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.ModuleProtocolandhk.SupportsCall. - Added
cross_replica_axistoVectorQuantiser. - Added
allow_reuseargument tohk.lift. - Added
fan_in_axestoVarianceScalinginitialiser. - 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_callimplementation. - The
HAIKU_FLATMAPPINGenv var is no longer used. hk.dropout(..)now supports dynamicrate.hk.without_apply_rng(..)now supports multi transformed functions.