rng (Array
) – jax.random key
n_dimensions (int
) – desired covariance dimensions
stdev_mean (float
) – standard deviation of location and log eigenvalues
(means for both are 0)
stdev_cov (float
) – standard deviated of the covariance
ridge (Union
[float
, Array
]) – Offset for means.