jax.random.maxwell#

jax.random.maxwell(key, shape=(), dtype=<class 'numpy.float64'>)[source]#

Sample from a one sided Maxwell distribution.

The scipy counterpart is scipy.stats.maxwell.

Parameters
  • key (Union[Array, PRNGKeyArray]) – a PRNG key.

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

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

Return type

Array

Returns

A jnp.array of samples, of shape shape.