install_pip_requirements.sh

pip install -U jaxlib==0.3.24+cuda11.cudnn82 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
pip install jax==0.3.25
pip install ml-collections==0.1.1
pip install dm-haiku==0.0.9
pip install pandas==1.4.4
pip install biopython==1.79
pip install chex==0.0.7
pip install dm-tree==0.1.6
pip install immutabledict==2.0.0
pip install numpy==1.19.5
pip install scipy==1.7.0
pip install tensorflow-cpu==2.12.0