minor: add @jit to threefry hash function in random.py
minor: add @jit to threefry hash function in random.py

PiperOrigin-RevId: 222841601

google/jax,tensorflow/probability,google/jax,google/jax,tensorflow/probability,google/jax