New style Jax RNG keys cannot be converted directly to Numpy random state seeds.

[miltonllera]

Jax recently started moving towards new style RNG-keys which entails changing calls to jax.random.PRNGKey for calls to jax.key. Additionally, seeds to numpy.random.RandomState from new-style keys (as in compute_cvt_centroids) must be first processed with jax.random.key_data otherwise we get a type error during conversion.

I am happy to write a pull request for this.

[Aneoshun]

Hi Milton,

Sounds great. Please, write de PR for this, we will be happy to review and accept it.