I wonder if this change has any implications for the TensorFlow vs. JAX situation/transition. IIRC I read that DeepMind mainly used JAX, but not sure about the Brain. Any insights from the people in the know? It seems JAX is the future, but TF dominates current production stacks.