Matt's trick / central formulation of beta distribution?

[junpenglao]

You can have a look at the recent paper Implicit Reparameterization Gradients, many of the tricks should also apply. Also see https://en.wikipedia.org/wiki/Beta_distribution#Related_distributions