4 ms·
Cupy was first but at this point you're better of using JAX. Has a much larger community, a big push from Google Research, and unlike PFN's Chainer (of which C
by towerwoerjrrwer 2y ago
Cupy was first but at this point you're better of using JAX.
Has a much larger community, a big push from Google Research, and unlike PFN's Chainer (of which CuPy is the computational base), is not semi-abandoned.
Kind of sad to see CuPy/Chainer eco-system die: not only did they pioneer the PyTorch programming model, but also stuck to Numpy API like JAX does (though the AD is layered on top in Chainer IIRC).
- p1esk 2y agoI agree. Though it's good to have options for GPU accelerated Numpy. Especially if Google decides to discontinue Jax at some point.
- WanderPanda 2y agoJAX still has the "This is a research project, not an official Google product. Expect bugs and sharp edges. Please help by trying it out, reporting bugs, and letting us know what you think!" disclaimer in its readme. This is quite scary, especially coming from Google which is known to abandon projects out of the blue
- atty 2y agoJax is, as far as an outsider can tell, Google research/DeepMind’s primary computation library now, not Tensorflow. So it’s safer than most Google projects, unless you think they’re secretly developing their third tensor computation library in 10 years (I suppose if anyone was going to do it, it would be Google).
- kmaehashi 2y agoCuPy isn't semi-abandoned as well, obviously :)
- low_tech_love 2y agoI tried Jax last year and was not impressed. Maybe it was just me, but everything I tried to do (especially with the autograd stuff) involved huge compilation times and I simply could not get the promised performance. Maybe I’ll try again and read some new documentation, since everyone is excited about it.