Это продолжение этого вопроса:

Как сделать сумму сумм квадрата суммы сумм?

Там, где я просил помочь использовать einsum (чтобы добиться значительного увеличения скорости), я получил отличный ответ.

Я также получилпредложение использоватьnumba, Я пошел и попробовал оба, и кажется, что после определенного момента увеличение скоростиnumba намного лучше.

Так как же это ускорить, не сталкиваясь с проблемами с памятью?