Both of these are pretty cool.

augment.py

That seems like something we could mention in the image data augmentation 5 tutorial. I’m sure some users end up on that page looking for RandAugment or another higher level function and are disappointed.