Reopen issue #599 (memory issue)

Question

Reopen issue #599 (memory issue)

PikaPei opened this issue 8 months ago · 3 comments

Hello,
I commented earlier on a different issue but couldn't reopen it, so I'm opening this one. Thank you.
#599 (comment)

Answer 1 · 2024-02-01T08:49:44.000Z

Thanks for opening this question. I also have no idea. Please do not assign any reference to Python variables if you do not need it. Make it an intermediate value and only return the needed analysis values.

Answer 2 · 2024-02-01T08:54:51.000Z

It seems that JAX has some strange memory leak issue. Please also see jax-ml/jax#11448, jax-ml/jax#12719 .

Answer 3 · 2024-02-01T09:17:55.000Z

Understood, thanks for your reply. I will now close this issue.