MSELoss

Auto-generated from jax_code_defs

PyTorch

API: torch.nn.functional.mse_loss
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: optax.l2_loss
Strategy: Plugin (loss_reduction)