If you are using a DataLoader with num_workers>0, multiple workers will be used to create each batch in the background.
I don’t know which lib would provide all functionality you need for the 3D data augmentation, but maybe a medical lib could be helpful, such as MONAI or torchio.