I think I may be out of luck for the time being–I’ve tried to run through the the different optimizers and their parameters, but haven’t gotten anything to run. I wonder if you’ve ever tried anything like splitting the dataset into batches and running multiple processes in parallel with pm.sample()? Of course, this assumes my model would sample successfully on datasets that could have slight underlying differences.