Get the latest tech news

I don't like NumPy


it’s too hard

To get all those transistors firing, you need to call special GPU functions that will sort of split up the arrays into lots of little pieces and process them in parallel. Life is short and precious, but if you spend a lot of yours reading the NumPy documentation, you might eventually realize that there’s a function called np.tensordot, and that it’s possible to make it do much of the work: The strings are a bit tedious, but they’re worth it, because the overall function is easy(ish) to understand, is completely explicit, and is quite general and powerful.

Get the Android app

Or read this on Hacker News

Read more on:

Photo of NumPy

NumPy

Related news:

News photo

Numpyro: Probabilistic programming with NumPy powered by Jax

News photo

CuPy: NumPy and SciPy for GPU

News photo

Ultra simplified "MNIST" in 60 lines of Python with NumPy