xla

0ee876f3 - Implement prng_key as a mutable array (#9305)