Thanks, @ricardoV94. I tried it got this error:
“ValueError: Random variables detected in the logp graph: [p, a, sigma_a, b, sigma_b, lambda, r, sigma_r, alpha, sigma_alpha].
This can happen when DensityDist logp or Interval transform functions reference nonlocal variables.”
I have attached the screenshot.