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 ]
There are no comments on this comment yet.