“Google's Pathways system supports both PyTorch and JAX, with JAX being the primary framework used internally at Google.”