CrossQ icon indicating copy to clipboard operation
CrossQ copied to clipboard

Published 20 hours ago • verified publisher icon adityab

JAX allocating too much memory

Open JankowskiChristopher opened this issue 10 months ago • 0 comments

Hello, In your code you correctly specified:

os.environ['XLA_PYTHON_CLIENT_PREALLOCATE'] = 'false'

However currently this line of code is after import jax therefore is not working. I fixed this in #5, so if you would like, you can fix this by merging the pr. I provided more information in #5.

JankowskiChristopher avatar Apr 12 '24 22:04 JankowskiChristopher