zero_grad

Auto-generated from jax_code_defs

PyTorchJAX (Core)Apple MLXFlax NNXPaxML / Praxis

PyTorch

API: optimizer.zero_grad

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: optimizer_zero_grad

Strategy: Plugin (optimizer_zero_grad)

Apple MLX

API: optimizer_zero_grad

Strategy: Plugin (mlx_zero_grad)

Flax NNX

API: —

Strategy: Plugin (optimizer_zero_grad)

PaxML / Praxis

API: —

Strategy: Plugin (optimizer_zero_grad)