Hi @Sascha, I believe you can use InMemoryDataset with vanilla PyTorch.
Datasets in PyTorch require only 3 functions [ __init__(), __len__(), __getitem__()]. Since torch_geometric.Dataset contains those 3 functions, you should be able to pass its subclasses to a vanilla PyTorch DataLoader.