Multi-chain init to value

Hi all, 
I'm trying to initialize my MCMC with one or more specific values for each chain. As numpyro.infer.util.init_to_value() doesn’t work for multiple chains and init_params argument takes only unconstrained values, it is possible to implement a function or something similar for multi-chain init to value ? 

Thanks a lot!