High-throughput tensor loading for PyTorch - View it on GitHub