News for hackers. Stuff that matters.

A comment on JAX – NumPy on the CPU, GPU, and TPU, with great automatic differentiation. Parent comment

Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati... by misja111 on Friday September 29, 2023 @04:41AM (#37700939)
> it'll run on a GPU for free and be about 1000x faster.

Are there any benchmarks for that? Running on GPU never comes for free. You have to transfer data back and forth which has a cost, for instance.

[ Parent | on HN ]
Re:JAX – NumPy on the CPU, GPU, and TPU, with great automati... by antognini on Friday September 29, 2023 @10:10AM (#37704533)
That was just from some quick benchmarks I did a few months back on some 10,000 particle N-body simulations. The performance boost will depend on the task, though. For the kinds of computations I did in grad school it would have been less effective since I was only looking at 3--5 objects, so there's just less parallelism to take advantage of.
[ Parent | on HN ]