PyTorch is where it's at, TensorFlow is just a mess of abstraction and complexity at this point.