no_grad ======= Auto-generated from jax_code_defs .. raw:: html

PyTorch

API: torch.no_grad
Strategy: Direct Mapping

JAX (Core)

API: contextlib.nullcontext
Strategy: Plugin (context_to_function_wrap)

Apple MLX

API: contextlib.nullcontext
Strategy: Plugin (context_to_function_wrap)