Fix #635
Created by: PhilipVinc
Closes #635 (closed) . Was an oversight on my part, as Numpy samplers are not jax structures. Probably could make them jax-compatible but this should work in the meantime.
Also fixed the docstring of log_value cc @gcarleo