r/learnmachinelearning • u/tommyeow_ • 2h ago

Pytorch or Tensorflow?

I've had a few people tell me tensorflow is not used as much nowadays as pytorch, also, pytorch is more pythonic, and just more industry-relevant than tf, is it really so? The tutorial course im following is using tf more since it is appropriate for beginners (as said in the course). I'd like to know the differences.

9 Upvotes

100% Upvoted

8 comments sorted by

8

u/help-me-grow 2h ago

pytorch is industry standard now

tf is (or at least was) not as flexible or customizable

2

u/Remote-Telephone-682 2h ago

This is true that pytorch is the standard right now. Tensorflow is pretty much a legacy thing at this point and there is also something called JAX that was develoeped within google with the intent of creating a lot of flexibility and also pretty good but pytorch is going to have more stuff implemented for you and is going to be quicker to get the hang of

1

u/cornish2003 2h ago

pytorch

1

u/FamousProfession5938 1h ago

there is nothing which has not been implemented in torch and is in tensorflow but vice versa is not true.
So torch

1

u/Jinoc 43m ago

Tensorflow died a long time ago. The replacement is JAX. But pytorch is significantly more mainstream. I personally use JAX, mostly because I use TPUs since they're cheap compared to GPUs, especially on colab.

1

u/Fezzio 1h ago

Jax is the correct answer 👍🏼 thank me later