Haiku Fundamentals — Haiku documentation
https://dm-haiku.readthedocs.io/en/latest/api.htmlParameters. f – A function closing over Module instances.. Return type. TransformedWithState. Returns. A TransformedWithState tuple with init and apply pure functions.. multi_transform¶ haiku.multi_transform (f) [source] ¶ Transforms a collection of functions using Haiku into pure functions. In many scenarios we have several modules which are used either as primitives for …