jax.random.maxwell

jax.random.maxwell(key, shape=(), dtype=<class 'float'>)

Sample from a one sided Maxwell distribution.

The values are distributed according to the probability density function:

\[f(x) \propto x^2 e^{-x^2 / 2}\]

on the domain \(0 \le x < \infty\).

Parameters:

• key (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – a PRNG key.

• shape (Sequence[int]) – The shape of the returned samples.

• dtype (Union[str, type[Any], dtype, SupportsDType]) – The type used for samples.

Return type:

Array

Returns:

A jnp.array of samples, of shape shape.