Creating a Dirichlet distribution from a random variable

You can further condense the code through using pm.math.sigmoid rather than defining it manually