Preparing Text Dataset

JAX doesn’t ship with data-loading utilities. This keeps JAX focused on providing a fast tool for building and training machine learning models. Loading data in JAX is done using either TensorFlow or PyTorch. This lesson will focus on how to load datasets in JAX using TensorFlow.

Let’s dive in!

Loading text data in JAX

Let’s use the Worldcup tweets dataset to illustrate how to load text datasets with JAX. We import the standard data science packages and then view a sample of the data.

Get hands-on with 1200+ tech skills courses.