Warning

This page was created from a pull request.

jax.random.maxwell

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

Sample from a one sided Maxwell distribution.

The scipy counterpart is scipy.stats.maxwell.

Parameters

• key – a PRNGKey key.

• shape – The shape of the returned samples.

• dtype – The type used for samples.

Returns

A jnp.array of samples, of shape shape.