Skip to content

Haiku 0.0.11

Compare
Choose a tag to compare
@tomhennigan tomhennigan released this 10 Nov 14:04
  • 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 supports dilation.
  • hk.dropout supports broadcast_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 mirroring jax.lax.map.