enable_grad

Auto-generated from jax_code_defs

PyTorchJAX (Core)Apple MLX

PyTorch

API: torch.enable_grad

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: contextlib.nullcontext

Strategy: Plugin (context_to_function_wrap)

Apple MLX

API: contextlib.nullcontext

Strategy: Plugin (context_to_function_wrap)