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

Thank you very much Ricardo! Deterministic variables indeed caused it. I was following the example of Discrete Choice and Random Utility Models and set p_ as deterministic variables. After replacing it with just the softmax function, the problem was solved!

1 Like