On: JAX – NumPy on the CPU, GPU, and TPU

[fulafel]

To spell out what the linked ISPC post implies, most of the difference, like ISPC shows, is differences in GPU languages and compilers vs CPU side equivalents.