IIRC pyro just imports constraints from pytorch.
Check constraint_registry 68 doc, there is a code example. Alternatively, you can just manually transform parameters, e.g. scale = F.softplus(self.raw_scale)