From 51849fc1874be9906f6439664917029dff74a7bd Mon Sep 17 00:00:00 2001
From: Dima <33123184+DimaMolod@users.noreply.github.com>
Date: Wed, 16 Oct 2024 16:33:14 +0200
Subject: [PATCH] install_jax

---
 .github/workflows/github_actions.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/github_actions.yml b/.github/workflows/github_actions.yml
index fccbce0e..4dc6028e 100644
--- a/.github/workflows/github_actions.yml
+++ b/.github/workflows/github_actions.yml
@@ -50,6 +50,7 @@ jobs:
           pip install -e .
           pip install -e ColabFold --no-deps
           pip install -e alphafold --no-deps
+          pip install -U "jax[cuda12]"
           export PYTHONPATH=$PWD/AlphaLink2:$PYTHONPATH
           # install dependencies for AlphaLink backend
           # pip install torch==1.13.0+cu117 --extra-index-url https://download.pytorch.org/whl/cu117