Poisson

Returns a tensor of the same size as input with each element sampled from a Poisson distribution.

Abstract Signature:

Poisson(input: Tensor, generator: Optional[Generator])

PyTorch

API: torch.poisson
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.random.poisson
Strategy: Plugin (inject_prng)

NumPy

API: numpy.random.poisson
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.random.poisson
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.random.poisson
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: —
Strategy: Custom / Partial

Flax NNX

API: jax.random.poisson
Strategy: Plugin (inject_prng)

PaxML / Praxis

API: jax.random.poisson
Strategy: Plugin (inject_prng)