CurrentUpdateContext ==================== Returns the current active UpdateContext for the given tag. **Abstract Signature:** ``CurrentUpdateContext(tag: Hashable)`` .. raw:: html

PyTorch

API:
Strategy: Custom / Partial

JAX (Core)

API:
Strategy: Custom / Partial

Flax NNX

API: flax.nnx.current_update_context
Strategy: Direct Mapping