probtorch / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Home Page:http://pytorch.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement Wishart distribution

alicanb opened this issue · comments

I think this will need multivariate polygamma

Task list:

  • Multivariate-polygamma
  • Distribution

Although slow, I think there can be a naive implementation for the multivariate polygamma functions correct? If the naive implementation is okay, I would be happy to take this up.

The naive implementation would be based on a summation of univariate polygamma functions.

I'm not sure it would be necessary. If you implement multivariate lgamma naively, autograd should take care of the rest? (Do take the multivariate lgamma up though...)

Ah, yes. Autograd should take up multivariate polygamma. In fact, we can use this as a testing mechanism. Shall I implement multivariate lgamma, if that's fine?