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