WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ProtocolError('Connection aborted ...

Hi. I am installing jax with cuda using venv and pip. I have found that the latest jax[cuda12] 0.5.3 packaging is somehow broken. If I install 0.5.3, 0.5.0, and then 0.5.3 again in the same ...