[luispedrocoelho]

"spending an hour figuring out what arcane incantation I need to pass to np.einsum to get the operation I want"

Yes, I have also had this experience and I hate how in the end, the code is very hard to read, while the for loop was probably trivial.

[cjalmeida]

Every time I thought I needed einsum or similar arcane ops, I found that a Numba optimization for loop did the job.

[gh02t]

Numba is nice, but it's another large dependency to pull in. If I can avoid it I will.