Viewing a single comment thread. View all comments

hcarlens OP t1_jb9q3cm wrote

Reply to comment by backhanderer in [R] Analysis of 200+ ML competitions in 2022 by hcarlens

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