operations = [
    Normalize(),
    RandomFlip(),
    grain.Batch(batch_size=64, drop_remainder=True),
]

        __
        __