def numba_vector_absolute_magnitude(crosses):
    """This function takes in an array of floats which is of size (M x N x 3). This
    represents N x M vectors, each with 3 components. The function then calculates
    and returns the vector absolute magnitude, which is the sum of the square of each
    component.

    Note: This function has been optimized for JIT compilation and parallel
    computation using Numba.

    Citation: Some or all of the following code was written by Jérôme Richard as a
    response to a question on Stack Overflow. The original response is here:
    https://stackoverflow.com/a/66757029/13240504.

    :param crosses: 3d array of floats
        This is the (M x N x 3) array, which holds the M x N vectors.
    :return: 2d array of floats
        This is the (M x N) array, which hold each vector's vector absolute magnitude.
    """
    return crosses[:, :, 0] ** 2 + crosses[:, :, 1] ** 2 + crosses[:, :, 2] ** 2