FYI, the error is the broadcasting in HalfNormal:
a.random(point=trace[0])
will generate the same error