Data Loading Maxtext supports input data pipelines in the following ways: Tf.data* Grain Hugging Face Datasets *Tf.data is the most performant way of loading large scale datasets. You can read more about the pipelines in .