3 ms·
Don't use TF though, go with PyTorch or maybe JAX.
by bigdict 3y ago
Don't use TF though, go with PyTorch or maybe JAX.
- kryptogeist 3y agoAs someone that has just started learning machine learning I can say that, in my experience, Pytorch is way more beginner friendly than TF. I had a hard time to setup TF in order for it work with my GPU. On the other hand pytorch was a breeze.
- snats 3y agoJAX feels like the new kid on the block. Every time that I see it is because it has made a project a loot faster (eg. WhisperJAX or the stable diffusion in JAX). Can anyone explain me why is it so fast is it because of parallelism?
- bigdict 3y agoBecause of XLA, a deep learning compiler that looks at the entire computational graph as a whole.