code: def mtf_image_transformer_cifar_4x():
    
    hparams = mtf_image_transformer_base_cifar()
    hparams.mesh_shape = "batch:32"
    hparams.layout = "batch:batch"
    hparams.batch_size = 128
    return hparams
docs: Data parallel CIFAR parameters.