SetAnomalyEnabled

Context-manager/Function to enable or disable anomaly detection (NaN checks).

Abstract Signature:

SetAnomalyEnabled(mode: bool)

PyTorchJAX (Core)TensorFlowApple MLX

PyTorch

API: torch.set_anomaly_enabled

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'jax.config.update('jax_debug_nans', {mode})'

TensorFlow

API: tf.debugging.enable_check_numerics

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial