Hi,
I am having the same problem of memory leakage. My input dataset is barely 200Mb.
Shape of dataset is 1.2M x 19.
I have used my local instance to run sampling. It would work fine on a smaller sample of the dataset. But when I plug everything in, it starts consuming more and more memory, until it starts swapping to disk and eventually kills the session.
I tried on heavy AWS Sagemaker compute but it does the same thing over there.
Using PYMC v5. Python 3.9.
Can anyone please help me on this?