Contents:
Randomizes the columns in X
key – a PRNG key used as the random key.
X (jax.numpy.ndarray) – A 2D data matrix
The data matrix with randomized columns
(jax.numpy.ndarray)