Normalizing flows in PyTorch - View it on GitHub