Closed bwpriest closed 1 year ago
We need to update our jax installation chassis and instructions to account for the current state of jax.
I think that I am just going to remove the GPU extras. If the user wants to use GPU JAX, they'll need to install it themselves.
Addressed with PR #140
We need to update our jax installation chassis and instructions to account for the current state of jax.