"Fast enough" parallel implementation of blake3 in cuda. Usage:

def checksum_any(x: jax.Array) -> bytes:
    x = jax.numpy.ravel(x)
    x = x.view(jax.numpy.uint8)
    return blake3_hash(x).tobytes()