needs lots of rng keys in the func

[JiahaoYao]

a silly question about jax: when one use jax to generate ur mcmc traj, one need the rng key for each step, right? how to handle that?

def mcmc(params, keys): 
		
		for i in range(20): 
			x = ..... + eps * random.normal(key_i, shape)
		
	return x

Do I need to pass the list of keys into this function?

[JiahaoYao]

okay, solved