Is Nx the equivalent of Numpy Python?

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