jax.extend.linear_util.transformation# jax.extend.linear_util.transformation = functools.partial(<class 'functools.partial'>, <function transformation>)[source]# Parameters: fun (WrappedFun) Return type: WrappedFun
