Ffcv: Train models at a fraction of the cost with accelerated data loading
github.com