Haiku 0.0.11
hk.layer_stack
now allows transparent application (no prefix on module names).hk.MultiHeadAttention
allows bias initializer to be configured or biases to be removed.hk.DepthwiseConvND
now supportsdilation
.hk.dropout
supportsbroadcast_dims
.hk.BatchApply
avoids an unnecessary h2d copy during tracing.hk.experimental.profiler_name_scopes
has been removed, these are on by default.- Added
hk.map
mirroringjax.lax.map
.