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

PyTorch

API: torch.nn.functional.huber_loss
Strategy: Direct Mapping

JAX (Core)

API: optax.huber_loss
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.losses.huber_loss
Strategy: Direct Mapping

Flax NNX

API: optax.huber_loss
Strategy: Direct Mapping