Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Warning
This page was created from a pull request (#9655).
Sample from a one sided Maxwell distribution.
The scipy counterpart is scipy.stats.maxwell.
key (Union[Any, PRNGKeyArray]) β a PRNG key.
Union
Any
PRNGKeyArray
shape (Sequence[int]) β The shape of the returned samples.
Sequence
int
dtype (Any) β The type used for samples.
ndarray
A jnp.array of samples, of shape shape.
previous
jax.random.logistic
next
jax.random.multivariate_normal