Imported from https://github.com/wsgharvey/pyro-infcomp-examples