code: def make_madry_ngpu(nb_classes=10, input_shape=(None, 28, 28, 1), **kwargs):
    
    layers = [Conv2DnGPU(32, (5, 5), (1, 1), "SAME"),
              ReLU(),
              MaxPool((2, 2), (2, 2), "SAME"),
              Conv2DnGPU(64, (5, 5), (1, 1), "SAME"),
              ReLU(),
              MaxPool((2, 2), (2, 2), "SAME"),
              Flatten(),
              LinearnGPU(1024),
              ReLU(),
              LinearnGPU(nb_classes),
              Softmax()]
  
    model = MLPnGPU(nb_classes, layers, input_shape)
    return model
docs: Create a multi-GPU model similar to Madry et al. (arXiv:1706.06083).