predict with alphafold meet jax problem

[Zhang-hn1125]

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'