flax

d5a25b77 - add lifted transforms to haiku migration guide