- [ ] Improvement with lazy shuffle and limit (i.e. sample) the product tuples in beforehand to have even more and nicer augmentations - [ ] implement the KNN strategy of the paper?