augmax

Augmax is a library for efficiently composable data augmentation on GPU using JAX. Version 0.4.1 supports JAX transformations and provides a functional, composable augmentation pipeline. Release cadence is occasional; primarily maintained on GitHub.