code: def mtf_resnet_tiny():
    
    hparams = mtf_resnet_base()
    hparams.num_layers = 2
    hparams.hidden_size = 64
    hparams.filter_size = 64
    hparams.batch_size = 16
    # data parallelism and model-parallelism
    hparams.col_blocks = 1
    hparams.mesh_shape = "batch:2"
    hparams.layout = "batch:batch"
    hparams.layer_sizes = [1, 2, 3]
    hparams.filter_sizes = [64, 64, 64]
    return hparams
docs: Catch bugs locally...