brainpy / BrainPy

Brain Dynamics Programming in Python
https://brainpy.readthedocs.io/
GNU General Public License v3.0
537 stars 94 forks source link

Reopen issue #599 (memory issue) #611

Closed PikaPei closed 10 months ago

PikaPei commented 10 months ago

Hello, I commented earlier on a different issue but couldn't reopen it, so I'm opening this one. Thank you. https://github.com/brainpy/BrainPy/issues/599#issuecomment-1918451756

chaoming0625 commented 10 months ago

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.

chaoming0625 commented 10 months ago

It seems that JAX has some strange memory leak issue. Please also see https://github.com/google/jax/issues/11448, https://github.com/google/jax/issues/12719 .

PikaPei commented 10 months ago

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