Haiku 0.0.8
- 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 todynamic_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
.