Is Nx the equivalent of Numpy Python?

I am trying to find the Elixir equivalent of Numpy:

I understand that https://github.com/elixir-nx/nx/tree/main/nx#readme is the direct equivalent.

Is this understanding correct?

Yes. Nx is a mixture of Numpy, JAX, and tensor serving.

7 Likes