Bernoulli

Draws binary random numbers (0 or 1) from a Bernoulli distribution.

Abstract Signature:

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

PyTorch

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

JAX (Core)

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

NumPy

API: np.random.binomial
Strategy: Direct Mapping
Official Docs ↗

Keras

API: —
Strategy: Custom / Partial

TensorFlow

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

Apple MLX

API: mlx.core.random.bernoulli
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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

PaxML / Praxis

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