examples/research_projects/jax-projects/big_bird/bigbird_flax.py/DataCollator/fetch_inputs
class DataCollator:    def fetch_inputs(self, input_ids: list):
        inputs = [self._fetch_inputs(ids) for ids in input_ids]
        return zip(*inputs)