GitHub repository: khdlr/augmax
Primary language: Python
Description: Efficiently Composable Data Augmentation on the GPU with Jax