What is you opinion about Tensorflow and Jax? Are they competitive currently? (For some reason PyTorch seems uncomfortable to use...)