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])`` .. raw:: html

PyTorch

API: torch.poisson
Strategy: Direct Mapping

JAX (Core)

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

NumPy

API: numpy.random.poisson
Strategy: Direct Mapping

Keras

API: keras.random.poisson
Strategy: Direct Mapping

TensorFlow

API: tf.random.poisson
Strategy: Direct Mapping

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)