+ set -e
+ pip install --no-warn-conflicts 'colabfold[alphafold-minus-jax]==1.6.1'
Collecting colabfold==1.6.1 (from colabfold[alphafold-minus-jax]==1.6.1)
  Downloading colabfold-1.6.1-py3-none-any.whl.metadata (21 kB)
Collecting appdirs<2.0.0,>=1.4.4 (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading appdirs-1.4.4-py2.py3-none-any.whl.metadata (9.0 kB)
Collecting biopython<1.86 (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading biopython-1.85-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (13 kB)
Collecting importlib-metadata<9.0.0,>=8.6.1 (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading importlib_metadata-8.9.0-py3-none-any.whl.metadata (4.5 kB)
Requirement already satisfied: matplotlib<4.0.0,>=3.8.4 in /usr/local/lib/python3.12/dist-packages (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (3.10.0)
Requirement already satisfied: numpy<3.0.0,>=2.0.2 in /usr/local/lib/python3.12/dist-packages (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2.0.2)
Requirement already satisfied: orjson<4.0.0,>=3.11.4 in /usr/local/lib/python3.12/dist-packages (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (3.11.9)
Requirement already satisfied: pandas<3.0.0,>=2.2.2 in /usr/local/lib/python3.12/dist-packages (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2.2.2)
Requirement already satisfied: requests<3.0.0,>=2.26.0 in /usr/local/lib/python3.12/dist-packages (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2.32.4)
Requirement already satisfied: tqdm<5.0.0,>=4.62.2 in /usr/local/lib/python3.12/dist-packages (from colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (4.67.3)
Requirement already satisfied: absl-py<2.0.0,>=1.0.0 in /usr/local/lib/python3.12/dist-packages (from colabfold[alphafold-minus-jax]==1.6.1) (1.4.0)
Collecting alphafold-colabfold==2.3.13 (from colabfold[alphafold-minus-jax]==1.6.1)
  Downloading alphafold_colabfold-2.3.13-py3-none-any.whl.metadata (1.3 kB)
Collecting dm-haiku<0.0.17,>=0.0.16 (from colabfold[alphafold-minus-jax]==1.6.1)
  Downloading dm_haiku-0.0.16-py3-none-any.whl.metadata (19 kB)
Requirement already satisfied: dm-tree<0.2.0,>=0.1.9 in /usr/local/lib/python3.12/dist-packages (from colabfold[alphafold-minus-jax]==1.6.1) (0.1.10)
Collecting py3Dmol<3.0.0,>=2.0.1 (from colabfold[alphafold-minus-jax]==1.6.1)
  Downloading py3dmol-2.5.5-py2.py3-none-any.whl.metadata (2.1 kB)
Collecting tensorflow-cpu<3.0.0,>=2.16.2 (from colabfold[alphafold-minus-jax]==1.6.1)
  Downloading tensorflow_cpu-2.21.0-cp312-cp312-manylinux_2_27_x86_64.whl.metadata (4.4 kB)
Collecting chex (from alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading chex-0.1.92-py3-none-any.whl.metadata (19 kB)
Requirement already satisfied: immutabledict in /usr/local/lib/python3.12/dist-packages (from alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1) (4.3.1)
Requirement already satisfied: jax in /usr/local/lib/python3.12/dist-packages (from alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1) (0.7.2)
Collecting ml-collections (from alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading ml_collections-1.1.0-py3-none-any.whl.metadata (22 kB)
Requirement already satisfied: scipy in /usr/local/lib/python3.12/dist-packages (from alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1) (1.16.3)
Collecting jmp>=0.0.2 (from dm-haiku<0.0.17,>=0.0.16->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading jmp-0.0.4-py3-none-any.whl.metadata (8.9 kB)
Requirement already satisfied: tabulate>=0.8.9 in /usr/local/lib/python3.12/dist-packages (from dm-haiku<0.0.17,>=0.0.16->colabfold[alphafold-minus-jax]==1.6.1) (0.9.0)
Requirement already satisfied: attrs>=18.2.0 in /usr/local/lib/python3.12/dist-packages (from dm-tree<0.2.0,>=0.1.9->colabfold[alphafold-minus-jax]==1.6.1) (26.1.0)
Requirement already satisfied: wrapt>=1.11.2 in /usr/local/lib/python3.12/dist-packages (from dm-tree<0.2.0,>=0.1.9->colabfold[alphafold-minus-jax]==1.6.1) (2.2.2)
Requirement already satisfied: zipp>=3.20 in /usr/local/lib/python3.12/dist-packages (from importlib-metadata<9.0.0,>=8.6.1->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (4.1.0)
Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (1.3.3)
Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (0.12.1)
Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (4.63.0)
Requirement already satisfied: kiwisolver>=1.3.1 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (1.5.0)
Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (26.2)
Requirement already satisfied: pillow>=8 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (11.3.0)
Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (3.3.2)
Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.12/dist-packages (from matplotlib<4.0.0,>=3.8.4->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2.9.0.post0)
Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.12/dist-packages (from pandas<3.0.0,>=2.2.2->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2025.2)
Requirement already satisfied: tzdata>=2022.7 in /usr/local/lib/python3.12/dist-packages (from pandas<3.0.0,>=2.2.2->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2026.2)
Requirement already satisfied: charset_normalizer<4,>=2 in /usr/local/lib/python3.12/dist-packages (from requests<3.0.0,>=2.26.0->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (3.4.7)
Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.12/dist-packages (from requests<3.0.0,>=2.26.0->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (3.18)
Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.12/dist-packages (from requests<3.0.0,>=2.26.0->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2.5.0)
Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.12/dist-packages (from requests<3.0.0,>=2.26.0->colabfold==1.6.1->colabfold[alphafold-minus-jax]==1.6.1) (2026.6.17)
Requirement already satisfied: astunparse>=1.6.0 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (1.6.3)
Requirement already satisfied: flatbuffers>=25.9.23 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (25.12.19)
Requirement already satisfied: gast!=0.5.0,!=0.5.1,!=0.5.2,>=0.2.1 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.7.0)
Requirement already satisfied: google_pasta>=0.1.1 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.2.0)
Requirement already satisfied: libclang>=13.0.0 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (18.1.1)
Requirement already satisfied: opt_einsum>=2.3.2 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (3.4.0)
Collecting protobuf<8.0.0,>=6.31.1 (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading protobuf-7.35.1-cp310-abi3-manylinux2014_x86_64.whl.metadata (595 bytes)
Requirement already satisfied: setuptools in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (75.2.0)
Requirement already satisfied: six>=1.12.0 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (1.17.0)
Requirement already satisfied: termcolor>=1.1.0 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (3.3.0)
Requirement already satisfied: typing_extensions>=3.6.6 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (4.15.0)
Requirement already satisfied: grpcio<2.0,>=1.24.3 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (1.81.1)
Requirement already satisfied: keras>=3.12.0 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (3.13.2)
Collecting h5py<3.15.0,>=3.11.0 (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading h5py-3.14.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (2.7 kB)
Requirement already satisfied: ml_dtypes<1.0.0,>=0.5.1 in /usr/local/lib/python3.12/dist-packages (from tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.5.4)
Requirement already satisfied: wheel<1.0,>=0.23.0 in /usr/local/lib/python3.12/dist-packages (from astunparse>=1.6.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.47.0)
Requirement already satisfied: rich in /usr/local/lib/python3.12/dist-packages (from keras>=3.12.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (13.9.4)
Requirement already satisfied: namex in /usr/local/lib/python3.12/dist-packages (from keras>=3.12.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.1.0)
Requirement already satisfied: optree in /usr/local/lib/python3.12/dist-packages (from keras>=3.12.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.19.1)
INFO: pip is looking at multiple versions of chex to determine which version is compatible with other requirements. This could take a while.
Collecting chex (from alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1)
  Downloading chex-0.1.91-py3-none-any.whl.metadata (18 kB)
  Downloading chex-0.1.90-py3-none-any.whl.metadata (18 kB)
Requirement already satisfied: jaxlib>=0.4.27 in /usr/local/lib/python3.12/dist-packages (from chex->alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1) (0.7.2)
Requirement already satisfied: toolz>=0.9.0 in /usr/local/lib/python3.12/dist-packages (from chex->alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1) (0.12.1)
Requirement already satisfied: PyYAML in /usr/local/lib/python3.12/dist-packages (from ml-collections->alphafold-colabfold==2.3.13->colabfold[alphafold-minus-jax]==1.6.1) (6.0.3)
Requirement already satisfied: markdown-it-py>=2.2.0 in /usr/local/lib/python3.12/dist-packages (from rich->keras>=3.12.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (4.2.0)
Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /usr/local/lib/python3.12/dist-packages (from rich->keras>=3.12.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (2.20.0)
Requirement already satisfied: mdurl~=0.1 in /usr/local/lib/python3.12/dist-packages (from markdown-it-py>=2.2.0->rich->keras>=3.12.0->tensorflow-cpu<3.0.0,>=2.16.2->colabfold[alphafold-minus-jax]==1.6.1) (0.1.2)
Downloading colabfold-1.6.1-py3-none-any.whl (79 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 79.8/79.8 kB 7.6 MB/s eta 0:00:00
Downloading alphafold_colabfold-2.3.13-py3-none-any.whl (248 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 248.4/248.4 kB 26.0 MB/s eta 0:00:00
Downloading appdirs-1.4.4-py2.py3-none-any.whl (9.6 kB)
Downloading biopython-1.85-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (3.3 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 3.3/3.3 MB 93.8 MB/s eta 0:00:00
Downloading dm_haiku-0.0.16-py3-none-any.whl (374 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 374.3/374.3 kB 33.3 MB/s eta 0:00:00
Downloading importlib_metadata-8.9.0-py3-none-any.whl (27 kB)
Downloading py3dmol-2.5.5-py2.py3-none-any.whl (7.2 kB)
Downloading tensorflow_cpu-2.21.0-cp312-cp312-manylinux_2_27_x86_64.whl (274.0 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 274.0/274.0 MB 5.4 MB/s eta 0:00:00
Downloading h5py-3.14.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.9 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.9/4.9 MB 131.3 MB/s eta 0:00:00
Downloading jmp-0.0.4-py3-none-any.whl (18 kB)
Downloading protobuf-7.35.1-cp310-abi3-manylinux2014_x86_64.whl (327 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 327.1/327.1 kB 35.7 MB/s eta 0:00:00
Downloading chex-0.1.90-py3-none-any.whl (101 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 101.0/101.0 kB 13.3 MB/s eta 0:00:00
Downloading ml_collections-1.1.0-py3-none-any.whl (76 kB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 76.7/76.7 kB 9.2 MB/s eta 0:00:00
Installing collected packages: py3Dmol, appdirs, protobuf, ml-collections, jmp, importlib-metadata, h5py, biopython, dm-haiku, colabfold, tensorflow-cpu, chex, alphafold-colabfold
  Attempting uninstall: protobuf
    Found existing installation: protobuf 5.29.6
    Uninstalling protobuf-5.29.6:
      Successfully uninstalled protobuf-5.29.6
  Attempting uninstall: importlib-metadata
    Found existing installation: importlib_metadata 9.0.0
    Uninstalling importlib_metadata-9.0.0:
      Successfully uninstalled importlib_metadata-9.0.0
  Attempting uninstall: h5py
    Found existing installation: h5py 3.16.0
    Uninstalling h5py-3.16.0:
      Successfully uninstalled h5py-3.16.0
Successfully installed alphafold-colabfold-2.3.13 appdirs-1.4.4 biopython-1.85 chex-0.1.90 colabfold-1.6.1 dm-haiku-0.0.16 h5py-3.14.0 importlib-metadata-8.9.0 jmp-0.0.4 ml-collections-1.1.0 protobuf-7.35.1 py3Dmol-2.5.5 tensorflow-cpu-2.21.0
+ pip install --upgrade dm-haiku
Requirement already satisfied: dm-haiku in /usr/local/lib/python3.12/dist-packages (0.0.16)
Requirement already satisfied: absl-py>=0.7.1 in /usr/local/lib/python3.12/dist-packages (from dm-haiku) (1.4.0)
Requirement already satisfied: jmp>=0.0.2 in /usr/local/lib/python3.12/dist-packages (from dm-haiku) (0.0.4)
Requirement already satisfied: numpy>=1.18.0 in /usr/local/lib/python3.12/dist-packages (from dm-haiku) (2.0.2)
Requirement already satisfied: tabulate>=0.8.9 in /usr/local/lib/python3.12/dist-packages (from dm-haiku) (0.9.0)
+ pip uninstall jax jaxlib -y
Found existing installation: jax 0.7.2
Uninstalling jax-0.7.2:
  Successfully uninstalled jax-0.7.2
Found existing installation: jaxlib 0.7.2
Uninstalling jaxlib-0.7.2:
  Successfully uninstalled jaxlib-0.7.2
+ pip install 'jax[cuda]==0.7.2' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Looking in links: https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Collecting jax==0.7.2 (from jax[cuda]==0.7.2)
  Downloading jax-0.7.2-py3-none-any.whl.metadata (13 kB)
Collecting jaxlib<=0.7.2,>=0.7.2 (from jax==0.7.2->jax[cuda]==0.7.2)
  Downloading jaxlib-0.7.2-cp312-cp312-manylinux_2_27_x86_64.whl.metadata (1.3 kB)
Requirement already satisfied: ml_dtypes>=0.5.0 in /usr/local/lib/python3.12/dist-packages (from jax==0.7.2->jax[cuda]==0.7.2) (0.5.4)
Requirement already satisfied: numpy>=2.0 in /usr/local/lib/python3.12/dist-packages (from jax==0.7.2->jax[cuda]==0.7.2) (2.0.2)
Requirement already satisfied: opt_einsum in /usr/local/lib/python3.12/dist-packages (from jax==0.7.2->jax[cuda]==0.7.2) (3.4.0)
Requirement already satisfied: scipy>=1.13 in /usr/local/lib/python3.12/dist-packages (from jax==0.7.2->jax[cuda]==0.7.2) (1.16.3)
Requirement already satisfied: jax-cuda12-plugin<=0.7.2,>=0.7.2 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (0.7.2)
Requirement already satisfied: jax-cuda12-pjrt==0.7.2 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin<=0.7.2,>=0.7.2->jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (0.7.2)
Requirement already satisfied: nvidia-cublas-cu12>=12.1.3.1 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.8.4.1)
Requirement already satisfied: nvidia-cuda-cupti-cu12>=12.1.105 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.8.90)
Requirement already satisfied: nvidia-cuda-nvcc-cu12>=12.6.85 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.8.93)
Requirement already satisfied: nvidia-cuda-runtime-cu12>=12.1.105 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.8.90)
Requirement already satisfied: nvidia-cudnn-cu12<10.0,>=9.8 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (9.19.0.56)
Requirement already satisfied: nvidia-cufft-cu12>=11.0.2.54 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (11.3.3.83)
Requirement already satisfied: nvidia-cusolver-cu12>=11.4.5.107 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (11.7.3.90)
Requirement already satisfied: nvidia-cusparse-cu12>=12.1.0.106 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.5.8.93)
Requirement already satisfied: nvidia-nccl-cu12>=2.18.1 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (2.28.9)
Requirement already satisfied: nvidia-nvjitlink-cu12>=12.1.105 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.8.93)
Requirement already satisfied: nvidia-cuda-nvrtc-cu12>=12.1.55 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (12.8.93)
Requirement already satisfied: nvidia-nvshmem-cu12>=3.2.5 in /usr/local/lib/python3.12/dist-packages (from jax-cuda12-plugin[with-cuda]<=0.7.2,>=0.7.2; extra == "cuda"->jax[cuda]==0.7.2) (3.4.5)
Downloading jax-0.7.2-py3-none-any.whl (2.8 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.8/2.8 MB 102.9 MB/s eta 0:00:00
Downloading jaxlib-0.7.2-cp312-cp312-manylinux_2_27_x86_64.whl (78.2 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 78.2/78.2 MB 10.9 MB/s eta 0:00:00
Installing collected packages: jaxlib, jax
Successfully installed jax-0.7.2 jaxlib-0.7.2
+ touch COLABFOLD_READY
