A better data loading pipeline for training ML models - View it on GitHub