Out of Memory during Transforming Variables Step using Numpyro. Increasing postprocessing_chunks doesn't seems to help. What should I do?

You might have very large deterministics. Can you conceivably fit them all in RAM at once?

1 Like