Solved by BramVanroy in post #2 
              
              
                I can highly recommend the huggingface datasets, which uses the very ast and efficient Apache arrow data format. GitHub - huggingface/datasets: 🤗 The largest hub of ready-to-use NLP datasets for ML models with fast, easy-to-use and efficient data manipulation tools