Is Nx the equivalent of Numpy Python?

I am trying to find the Elixir equivalent of Numpy:

I understand that nx/nx at main · elixir-nx/nx · GitHub is the direct equivalent.

Is this understanding correct?

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