WTTAT

is CPU acceleration available in this repo?

3

Here is the paper: 4.2.1 CPU acceleration. To accelerate the CPU stage, three independent sequential MSA searches can be arranged in parallel (Fig 4). Due to the limited CPU cores...

jax._src.traceback_util.UnfilteredStackTrace: AttributeError: module 'jax.dtypes' has no attribute 'prng_key'

13

i am running alphafold v2.3.2 to predict a multimer on 16CPU,241G,4* V100 ``` >3JA9_1|Chains A|Proliferating cell nuclear antigen|Homo sapiens (9606) MFEARLVQGSILKKVLEALKDLINEACWDISSSGVNLQSMDSSHVSLVQLTLRSEGFDTYRCDRNLAMGVNLTSMSKILKCAGNEDIITLRAEDNADTLALVFEAPNQEKVSDYEMKLMDLDVEQLGIPEQEYSCVVKMPSGEFARICRDLSHIGDAVVISCAKDGVKFSASGELGNGNIKLSQTSNVDKEEEAVTIEMNEPVQLTFALRYLNFFTKATPLSSTVTLSMSADVPLVVEYKIADMGHLKYYLAPKIEDEEGS >3JA9_2|Chains B|Proliferating cell nuclear antigen|Homo sapiens (9606) MFEARLVQGSILKKVLEALKDLINEACWDISSSGVNLQSMDSSHVSLVQLTLRSEGFDTYRCDRNLAMGVNLTSMSKILKCAGNEDIITLRAEDNADTLALVFEAPNQEKVSDYEMKLMDLDVEQLGIPEQEYSCVVKMPSGEFARICRDLSHIGDAVVISCAKDGVKFSASGELGNGNIKLSQTSNVDKEEEAVTIEMNEPVQLTFALRYLNFFTKATPLSSTVTLSMSADVPLVVEYKIADMGHLKYYLAPKIEDEEGS >3JA9_3|Chains...