3 ms·
Realistically you should use Numpy or Cupy (or whatever the appropriate/fashionable library is) anyway, because tuning this stuff is a big pain. So, GPUs have
by bee_rider 2y ago
Realistically you should use Numpy or Cupy (or whatever the appropriate/fashionable library is) anyway, because tuning this stuff is a big pain.
So, GPUs have the slight disadvantages that you have to think an about data movement and the drivers are a little less convenient to install, but it isn’t really a big deal.
- lmeyerov 2y agoAgreed! The bigger shift is switching to data parallel coding styles.
- PartiallyTyped 2y agoI am a big fan of jax for numerical computations these days.
- bee_rider 2y agoI’ve been seeing lots of posts about it lately. Haven’t had a chance to try it out, though.