augmax (pypi): Efficiently Composable Data Augmentation on the GPU with Jax
Keywords: gpu|jax|pre-processing