I'm in process of refactoring probtorch to use pytorch distributions, Kumaraswamy is the only distribution that we have missing in pytorch.