jax.extend.linear_util module#
WrappedFun(f, f_transformed, transforms, ...)
Represents a function f to which transforms are to be applied.
cache(call, *[, explain])
Memoization decorator for functions taking a WrappedFun as first argument.
merge_linear_aux(aux1, aux2)
wrap_init(f[, params, debug_info])
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3