Hi, TF maintainer here! We’ve been working on trying to add some helper functions for this. For example, the recent notebooks use a to_tf_dataset() method that can be used to convert HuggingFace datasets to TF Dataset objects, which allows you to pass them straight to Keras methods like model.fit(). This avoids the need for separate TF data handling code in a lot of cases.