Felix Friedrich

Results 5 comments of Felix Friedrich

I got the same error, but running only: `pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html` worked for me, so it's probably because of the dm_haiku problem as described above

I'm experiencing the same issue with the BLOOM models

Same for me. Everything works on 0.7.6 now, and before it didn't. However, 0.8.0. does not resolve the issue and gives similar behavior as the others showed.

great functionality!