Learned Compression of High Dimensional Image Datasets