Skip to content

Haiku 0.0.8

Compare
Choose a tag to compare
@tomhennigan tomhennigan released this 21 Sep 14:10
  • Added experimental.force_name.
  • Added ability to simulate a method name in experimental.name_scope.
  • Added a config option for PRNG key block size.
  • Added unroll parameter to dynamic_unroll.
  • Remove use of deprecated jax.tree_* functions.
  • Many improvements to our examples.
  • Improve error messages in vmap.
  • Support jax_experimental_name_stack in jaxpr_info.
  • transform_and_run now supports a map on PRNG keys.
  • remat now uses the new JAX remat implementation.
  • Scale parameter is now optional in RMSNorm.