I am trying to upgrade my current tensorflow-metal (v.0.3.0) to 0.5.0. Does anyone know how to do it with "pip install -upgrade ...." command in the terminal?

Hi all, I couldn't use random.PRNGKey to generate random seed. Wondering anyone has similar issue before and figure this out. Here is my current config: jax-metal==0.0.3, jaxlib==0.4.10, jax==0.4.11. I am using Apple M1 pro.