Truncated and folded distributions (Links to Numpyro docs)