examples/research_projects/jax-projects/hybrid_clip/run_hybrid_clip.py/ImageTextDataset/__getitem__
class ImageTextDataset:    def __getitem__(self, index: int):
        image = self._load_image(index)
        target = self._load_target(index)

        if self.transforms is not None:
            image, target = self.transforms(image, target)

        return image, target