Restrict sampler to only draw from the prior for a particular variable

Replace _xi by a pm.Normal. It still needs to be a proper model variable for the rest to work.

Small note: you seem to be mixing variance with standard deviation in your custom step sampler