RosettaCommons/RFDesign

predict with alphafold meet jax problem

Zhang-hn1125 opened this issue · 0 comments

When I want to run the script af2_metrics.py, I get the following ModuleNotFoundError:
if jax.config.jax_experimental_name_stack and module_name:
AttributeError: 'Config' object has no attribute 'jax_experimental_name_stack'