jax

5c89170d - jaxlib: Fix Python 3.9 build and drop CUDA 10.0 support

jaxlib: Fix Python 3.9 build and drop CUDA 10.0 support The TF/XLA build no longer works with CUDA 10.0. This could potentially be fixed, but isn't easy or officially supported.