IsGradEnabledΒΆ
Returns True if grad mode is currently enabled.
JAX (Core)
API:
βStrategy: Macro 'True'
NumPy
API:
βStrategy: Macro 'False'
Keras
API:
βStrategy: Macro 'True'
TensorFlow
API:
βStrategy: Macro 'tf.executing_eagerly()'
Apple MLX
API:
βStrategy: Macro 'True'
Flax NNX
API:
βStrategy: Macro 'True'
PaxML / Praxis
API:
βStrategy: Macro 'True'