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.
