OK, I think I found it here: Probabilistic Matrix Factorization for Making Personalized Recommendations — PyMC3 3.10.0 documentation