EnableGrad ========== Context-manager that enables gradient calculation. .. raw:: html

PyTorch

API: torch.enable_grad
Strategy: Direct Mapping

JAX (Core)

API: contextlib.nullcontext
Strategy: Direct Mapping

Keras

API: None
Strategy: Direct Mapping

TensorFlow

API: tf.GradientTape
Strategy: Direct Mapping

Flax NNX

API: contextlib.nullcontext
Strategy: Direct Mapping