jax.extend.linear_util.wrap_init# jax.extend.linear_util.wrap_init(f, params=None, *, debug_info=None)[source]# Parameters: f (Callable) Return type: WrappedFun