jax.extend.linear_util 模組

StoreException
WrappedFun(f, f_transformed, transforms, ...)	表示要對其套用 transforms 的函式 f。
cache(call, *[, explain])	用於將 WrappedFun 作為第一個引數的函式的記憶化裝飾器。
merge_linear_aux(aux1, aux2)
轉換
帶輔助資訊的轉換
wrap_init(f[, params, debug_info])	將函式 f 包裝為 WrappedFun，適用於轉換。