Creator of Keras confirmed that the new version comes out in a few days. Keras becomes multi-backend again with support for PyTorch, TensorFlow and JAX. Personally, I’m excited to be able to try JAX without having to deep dive into documentation and entire ecosystem. What about you?
Actually, another perspective to put is that TensorFlow’s deployment is something JAX doesn’t have (not that I know of) and cutting it would be idiotic for google, since they eliminated their own tool in an ongoing AI revolution. TensorFlow is their current tool and if they are going to abandon it, they will need a strong replacement for it’s deployability which does guarantee a few years (since the JAX team doesn’t seem to be quite focused in deployment). IIRC JAX deploys by Tensorflow rn.