3 ms·
re: leading comparison, that makes sense. I recommend labeling it a bit better: JAX on TPU vs NumPy on CPU
by brrrrrm 5y ago
re: leading comparison, that makes sense. I recommend labeling it a bit better: JAX on TPU vs NumPy on CPU
- SleekEagle 5y agoUpdated! Thanks for the feedback. Added this note in the figure description "(n.b. JAX is using TPU and NumPy is using CPU in order to highlight that JAX's speed ceiling is much higher than NumPy's)"
- p1esk 5y agoComparing Jax on TPU vs Numpy on CPU does not make any sense. Of course a GEMM hardware accelerator will be much faster than a general purpose CPU. What is the point of this comparison? You either run the same code on different hardware, or different code on the same hardware. A much more interesting comparison would be CuPy or Pytorch code vs Jax code running on A100.