EnableGrad

Context-manager that enables gradient calculation.

PyTorch

API: torch.enable_grad
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: contextlib.nullcontext
Strategy: Direct Mapping
Official Docs ↗

Keras

API: None
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.GradientTape
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: contextlib.nullcontext
Strategy: Direct Mapping
Official Docs ↗