IsInferenceModeEnabled

Returns True if inference mode is currently enabled.

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.is_inference_mode_enabled

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'False'

NumPy

API: —

Strategy: Macro 'False'

Keras

API: —

Strategy: Macro 'False'

TensorFlow

API: —

Strategy: Macro 'False'

Apple MLX

API: —

Strategy: Macro 'False'

Flax NNX

API: —

Strategy: Macro 'False'

PaxML / Praxis

API: —

Strategy: Macro 'False'