examples/research_projects/jax-projects/hybrid_clip/run_hybrid_clip.py/ImageTextDataset/_load_image
class ImageTextDataset:    def _load_image(self, idx: int):
        path = self.image_paths[idx]
        return read_image(path, mode=ImageReadMode.RGB)