Pymc/numpyro GPU memory allocation

Thanks. If I only want the posterior, what else should I cut out through proper argument selection?