examples/research_projects/jax-projects/dataset-streaming/run_mlm_flax_stream.py/tokenize_function
    def tokenize_function(examples):
        return tokenizer(examples[data_args.text_column_name], return_special_tokens_mask=True)