Replace all remaining jnp.* calls with backend.numpy.* calls (#710)

Replace all remaining jnp.* calls with backend.numpy.* calls (#710) #1229