nahyup
Results
1
comments of
nahyup
I think that jaxlib version may cause the segfault. I suffered same issue. "pip install jaxlib==0.1.71+cuda111 -f https://storage.googleapis.com/jax-releases/jax_releases.html" may solve your problem