Have you checked out numba? You can use regular Python for loops and numpy calls and it will compile it. Instead of numpy vectorization mess.