The helper function below uses a seed to initialize a random number generator. As long as you use the same seed, you'll get the exact same results. Feel free to use different seeds when exploring results later in the guide.  
```python
def create_key(seed=0):
return jax.random.PRNGKey(seed)
```  
The helper function, or `rng`, is split 8 times so each device receives a different generator and generates a different image.  
```python
rng = create_key(0)
rng = jax.random.split(rng, jax.device_count())
```