IsAutocastCpuEnabled

Returns True if autocast is enabled on CPU.

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.is_autocast_cpu_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'