A demo implementation of Multi-Layer Approximate Message Passing applied to a trained generative neural network. The algorithm is introduced by Pandit et al. 2019 and their original implementation is here.

This implementation is written in Pytorch and it supports generative networks using Conv2D, Upsample (bilinear upsampling), ReLU, and Tanh layers.