How to use jax.random package for sampling general normal distribution?

[EngineerKhan]

While jax.random.normal() gives standard normal distribution. How can we use the jax.random package to sample a non-standard normal distribution (without using the shifting property, etc.) like we do in Numpy?

[jakevdp]

The equivalent of

x = np.random.normal(mean, std, shape)

would be

x = mean + std * jax.random.normal(key, shape)