We develop an efficient alternative to PyTorch built-in dataloader class for the case of in-memory datasets, and lightweight models.| Alex Shtoff