Viewing a single comment thread. View all comments

backhanderer t1_jb9ph2v wrote

Thanks for this. I knew PyTorch was dominant but didn’t realise it was this dominant for deep learning!

75

hcarlens OP t1_jb9q3cm wrote

Yeah, not just competitive ML but the research community as a whole seem to have almost entirely switched to PyTorch now (based on the Papers With Code data). I was expecting to see some people using JAX though!

37

TubasAreFun t1_jbapwmb wrote

JAX does offer some general matrix math that can be more useful/fast than torch alone. I often do deep learning with torch and then use JAX on the top to train statistical models (i.e. fuse features from multiple models, raw features, etc. into a single regression/inference)

16

saintshing t1_jbdbgwy wrote

Is pytorch also better than TF for usecases where I have to do training/inference on mobile?

2