Everyone uses PyTorch both in industry and academic research. Some labs use JAX/Flax. Companies that ships ML products often use Tensorflow. I want to say th

Everyone uses PyTorch both in industry and academic research. Some labs use JAX/Flax. Companies that ships ML products often use Tensorflow. I want to say that it doesn't matter much which one you start with, but you will have to learn PyTorch anyway- in the future- if you are serious about DL. And the experience of working with PyTorch is Astronomical Units better than Tensorflow. PyTorch is much more pleasant to work with and you can do really custom stuff pretty easily.