Consider moving to a third-party library to handle datasets (e.g. the huggingface datasets library)

Consider moving to a third-party library to handle datasets (e.g. the huggingface datasets library) #137