A RetroSearch Logo

Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Search Query:

Showing content from https://docs.jax.dev/en/latest/_autosummary/jax.extend.linear_util.cache.html below:

jax.extend.linear_util.cache — JAX documentation

jax.extend.linear_util.cache#
jax.extend.linear_util.cache(call, *, explain=None)[source]#

Memoization decorator for functions taking a WrappedFun as first argument.

Parameters:
  • call (Callable) – a Python callable that takes a WrappedFun as its first argument. The underlying transforms and params on the WrappedFun are used as part of the memoization cache key.

  • explain (Callable[[WrappedFun, bool, dict, tuple, float], None] | None) – a function that is invoked upon cache misses to log an explanation of the miss. Invoked with (fun, is_cache_first_use, cache, key, elapsed_sec).

Returns:

A memoized version of call.


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.4