#### Fixes Sometimes, memory issues lead to this error. Try ```bash XLA_PYTHON_CLIENT_PREALLOCATE=False ``` or investigate memory usage. Otherwise, there might be a mixed installation of [[CUDA]] libraries or a missmatch between these libraries and the NVIDIA driver in use. >[!bug] JAX 0.6.1 >A known [[- JAX -|JAX]] issue arises in version `0.6.1`: >An incompatibility between `nvidia-cusparse-cu12` and `nvidia-sublas-cu12` that is caused by an upper limit in the version JAX installs. To solve, install as usual and then manually update >```bash >pip isntall nvidia-cublas-cu12=12.9.0.13 >```