Replies: 1 comment
-
Hi - can you say more about what your "typical jax script" includes? This warning (and the deadlock that it warns about) will occur if your script attempts to use JAX together with multiprocessing. JAX's computational model is not compatible with multiprocessing. Is it possible your script is attempting to use JAX with some library like |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I run a typical
jax
script and this warning pops upafter running
import jax; jax.print_environment_info()
I getThis was after a fresh reinstall in a new virtual environment. Has anyone got any pointers here? I assume this works for other people and I'm doing something wrong!
Beta Was this translation helpful? Give feedback.
All reactions