PoissonNllLoss

Poisson Negative Log Likelihood Loss.

Abstract Signature:

PoissonNllLoss(input: Tensor, target: Tensor, log_input: bool = True, full: bool = False, eps: float = 1e-08, reduction: str = mean)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.poisson_nll_loss

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: optax.poisson_nll_loss

Strategy: Plugin (loss_reduction)

NumPy

API: —

Strategy: Custom / Partial

Keras

API: keras.losses.poisson

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.nn.log_poisson_loss

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: optax.poisson_nll_loss

Strategy: Plugin (loss_reduction)

PaxML / Praxis

API: optax.poisson_nll_loss

Strategy: Plugin (loss_reduction)