Add missing JAX samplers

Created by: gcarleo

There is a few missing samplers in sampler.jax.*