jax.extend.random.threefry_2x32

Contents

jax.extend.random.threefry_2x32#

jax.extend.random.threefry_2x32(keypair, count)[source]#

Apply the Threefry 2x32 hash.

Parameters:
  • keypair – a pair of 32bit unsigned integers used for the key.

  • count – an array of dtype uint32 used for the counts.

Returns:

An array of dtype uint32 with the same shape as count.