Joint graph learning and model fitting in Laplacian Regularized Stratified Models - View it on GitHub